diff --git a/notebooks/lab99_colab_testing.ipynb b/notebooks/lab99_colab_testing.ipynb index d38ec92..70edcc5 100644 --- 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",