Fix pytorch environment for package sync (#48)

pytorch.ipynb

@@ -118,7 +118,7 @@
     "    #  Confirm that GPU shows up\n",
     "    if torch.cuda.is_available():\n",
     "        device = \"cuda\"\n",
-    "        print(\"Using GPU {torch.cuda.get_device_name(torch.cuda.current_device())} 😎\\n\")\n",
+    "        print(f\"Using GPU {torch.cuda.get_device_name(torch.cuda.current_device())} 😎\\n\")\n",
     "    else:\n",
     "        device = \"cpu\"\n",
     "        print(\"Using CPU 😔\\n\")\n",
@@ -217,13 +217,21 @@
     "\n",
     "model = train_on_gpu()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b6e7ae53-ce21-49f3-856a-87774a19bf37",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:pytorch]",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "conda-env-pytorch-py"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -235,7 +243,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,

pytorch.yml

@@ -2,7 +2,7 @@ name: pytorch
 channels:
   - conda-forge
 dependencies:
-  - python=3.10
+  - python=3.11
   - dask
   - coiled
   - ipykernel
@@ -12,7 +12,6 @@ dependencies:
   - optuna
   - pytorch
   - torchvision
-  - pynvml
   - matplotlib
   - pip
   - optuna-integration