updates dependencies -- CUDA, torchvision, seaborn

the-full-stack · Jan 17, 2024 · ac59bfe · ac59bfe
1 parent 8f9455d
commit ac59bfe
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/notebooks/lab99_colab_testing.ipynb b/notebooks/lab99_colab_testing.ipynb
@@ -149,7 +149,7 @@
         "import torch\n",
         "\n",
         "assert \"2.1.\" in torch.__version__, torch.__version__\n",
-        "assert \"cu118\" in torch.__version__, torch.__version__\n",
+        "assert \"cu121\" in torch.__version__, torch.__version__\n",
         "torch.__version__"
       ],
       "metadata": {
@@ -161,7 +161,7 @@
     {
       "cell_type": "code",
       "source": [
-        "assert torch.version.cuda == \"11.8\", torch.version.cuda\n",
+        "assert torch.version.cuda == \"12.1\", torch.version.cuda\n",
         "torch.version.cuda"
       ],
       "metadata": {
@@ -174,7 +174,7 @@
       "cell_type": "code",
       "source": [
         "cudnn_version = \".\".join(str(v) for v in torch._C._cudnn.getRuntimeVersion())\n",
-        "assert \"8.7.\" in cudnn_version, cudnn_version\n",
+        "assert \"8.9.\" in cudnn_version, cudnn_version\n",
         "cudnn_version"
       ],
       "metadata": {
@@ -189,7 +189,7 @@
         "import torchvision\n",
         "\n",
         "assert \"0.16.\" in torchvision.__version__, torchvision.__version__\n",
-        "assert \"cu118\" in torchvision.__version__, torchvision.__version__\n",
+        "assert \"cu121\" in torchvision.__version__, torchvision.__version__\n",
         "torchvision.__version__"
       ],
       "metadata": {
@@ -300,8 +300,7 @@
       "source": [
         "import seaborn as sns\n",
         "\n",
-        "assert \"0.12\" in sns.__version__, sns.__version__\n",
-        "\n",
+        "assert \"0.13\" in sns.__version__, sns.__version__\n",
         "sns.__version__"
       ],
       "metadata": {
@@ -348,7 +347,7 @@
         "pytest==7.1.1\n",
         "pytest-cov==3.0.0\n",
         "# versioned to match Google Colab\n",
-        "seaborn>=0.12,<0.13\n",
+        "seaborn>=0.13,<0.14\n",
         "tornado>=6.3,<6.4\n",
         "# versioned to improve stability\n",
         "pytorch-lightning==1.6.3\n",