mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] MMSegmentation Tutorial running on colab error fix (#1452)

Browse Source 
* 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
This commit is contained in:
Andrey 2022-04-06 16:04:37 +02:00 committed by GitHub
parent 4bc2a30ea0
commit 9619deddef
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo/MMSegmentation_Tutorial.ipynb
View File

@ -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"
]