Skip to content

Commit

Permalink
[Fix] MMSegmentation Tutorial running on colab error fix (#1452)
Browse files Browse the repository at this point in the history
* 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 Apr 6, 2022
1 parent 3432ea9 commit 4bd32f4
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 4bd32f4

Please sign in to comment.