From 4bd32f4c3dec15b54e64197570edc3ab3fdcf164 Mon Sep 17 00:00:00 2001 From: Andrey Date: Wed, 6 Apr 2022 16:04:37 +0200 Subject: [PATCH] [Fix] MMSegmentation Tutorial running on colab error fix (#1452) * 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. * Update description about versions used --- demo/MMSegmentation_Tutorial.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/MMSegmentation_Tutorial.ipynb b/demo/MMSegmentation_Tutorial.ipynb index 9cefa399c..4e2ed56e1 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. " ] }, { @@ -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" ]