From 3ef2786a289a6c8129b622e150e7d96eaa79fa20 Mon Sep 17 00:00:00 2001 From: Andrey Date: Wed, 6 Apr 2022 15:28:21 +0200 Subject: [PATCH 1/2] update pytorch and cudatoolkit versions for colab Previous version was throwing an error after calling for mmseg.apis import (line 147) due to incompatible cudatoolkit version in colab. --- demo/MMSegmentation_Tutorial.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/MMSegmentation_Tutorial.ipynb b/demo/MMSegmentation_Tutorial.ipynb index 9cefa399c1..7e1076ec22 100644 --- a/demo/MMSegmentation_Tutorial.ipynb +++ b/demo/MMSegmentation_Tutorial.ipynb @@ -67,7 +67,7 @@ "outputs": [], "source": [ "# Install PyTorch\n", - "!conda install pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch\n", + "!conda install pytorch=1.10.0 torchvision cudatoolkit=11.1 -c pytorch\n", "# Install MMCV\n", "!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6/index.html" ] From bd299f610ef4a296b3217e69de3c0860213aa628 Mon Sep 17 00:00:00 2001 From: Andrey Date: Wed, 6 Apr 2022 15:29:31 +0200 Subject: [PATCH 2/2] Update description about versions used --- demo/MMSegmentation_Tutorial.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/MMSegmentation_Tutorial.ipynb b/demo/MMSegmentation_Tutorial.ipynb index 7e1076ec22..4e2ed56e1e 100644 --- a/demo/MMSegmentation_Tutorial.ipynb +++ b/demo/MMSegmentation_Tutorial.ipynb @@ -33,7 +33,7 @@ "## Install MMSegmentation\n", "This step may take several minutes. \n", "\n", - "We use PyTorch 1.6 and CUDA 10.1 for this tutorial. You may install other versions by change the version number in pip install command. " + "We use PyTorch 1.10 and CUDA 11.1 for this tutorial. You may install other versions by change the version number in pip install command. " ] }, {