Skip to content

Commit 4bd32f4

Browse files
authored
[Fix] MMSegmentation Tutorial running on colab error fix (open-mmlab#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
1 parent 3432ea9 commit 4bd32f4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

demo/MMSegmentation_Tutorial.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
"## Install MMSegmentation\n",
3434
"This step may take several minutes. \n",
3535
"\n",
36-
"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. "
36+
"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. "
3737
]
3838
},
3939
{
@@ -67,7 +67,7 @@
6767
"outputs": [],
6868
"source": [
6969
"# Install PyTorch\n",
70-
"!conda install pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch\n",
70+
"!conda install pytorch=1.10.0 torchvision cudatoolkit=11.1 -c pytorch\n",
7171
"# Install MMCV\n",
7272
"!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6/index.html"
7373
]

0 commit comments

Comments
 (0)