From d8cf381718f1fdead3fff3c74444079b14b76b30 Mon Sep 17 00:00:00 2001
From: James Bourbeau <jrbourbeau@gmail.com>
Date: Thu, 16 May 2024 14:50:58 -0500
Subject: [PATCH 1/2] Fix pytorch environment for package sync

---
 pytorch.yml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pytorch.yml b/pytorch.yml
index 12a1af2..e833db8 100644
--- a/pytorch.yml
+++ b/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

From b7b0ab5f70b74710177267b0dea68a73d501d6e9 Mon Sep 17 00:00:00 2001
From: James Bourbeau <jrbourbeau@gmail.com>
Date: Thu, 16 May 2024 14:59:19 -0500
Subject: [PATCH 2/2] f

---
 pytorch.ipynb | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/pytorch.ipynb b/pytorch.ipynb
index 9aa26b0..0b46c37 100644
--- a/pytorch.ipynb
+++ b/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,