Skip to content

Commit

Permalink
[Fix] MMSegmentation Tutorial running on colab error fix (open-mmlab#…
Browse files Browse the repository at this point in the history
…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
  • Loading branch information
androbaza authored and mob5566 committed Apr 13, 2022
1 parent d955f3c commit fe71e6f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions demo/MMSegmentation_Tutorial.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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. "
]
},
{
Expand Down Expand Up @@ -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"
]
Expand Down

0 comments on commit fe71e6f

Please sign in to comment.