|
1 | 1 | echo "Installing icevision + dependencices for CUDA 10"
|
| 2 | +echo "Uninstalling some dependencies to prevent errors" |
| 3 | +pip uninstall torchvision -y |
| 4 | +pip uninstall fastai -y |
| 5 | + |
| 6 | + echo "Installing some dependencies to prevent errors" |
| 7 | +pip install PyYAML>=5.1 -U -q |
| 8 | +pip install datascience -U -q |
| 9 | +pip install tensorflow==2.4.0 -U -q |
| 10 | +pip install google-colab -U -q |
| 11 | + |
| 12 | +echo "- Installing torch and its dependencies" |
2 | 13 | echo "- Installing torch and its dependencies"
|
3 |
| -pip install torchtext==0.9.0 --upgrade -q |
4 |
| -pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html --upgrade -q |
| 14 | +pip install torchtext==0.9.0 -U -q |
| 15 | +pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html -U -q |
| 16 | + |
5 | 17 | echo "- Installing fastai"
|
6 |
| -pip install fastai==2.3.1 --upgrade -q |
| 18 | +pip install fastai==2.3.1 -U -q |
| 19 | + |
7 | 20 | echo "- Installing icevision from master"
|
8 |
| -pip install git+git://github.com/airctic/icevision.git\#egg=icevision[all] --upgrade -q |
| 21 | +pip install git+git://github.com/airctic/icevision.git@fix-colab-install -U -q |
9 | 22 | echo "- Installing icedata from master"
|
10 |
| -pip install git+git://github.com/airctic/icedata.git --upgrade -q |
| 23 | +pip install git+git://github.com/airctic/icedata.git -U -q |
11 | 24 | echo "- Installing yolov5-icevision"
|
12 |
| -pip install yolov5-icevision --upgrade -q |
| 25 | +pip install yolov5-icevision -U -q |
13 | 26 | echo "- Installing mmcv"
|
14 |
| -pip install mmcv-full==1.3.7 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html --upgrade -q |
| 27 | +pip install mmcv-full==1.3.7 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html -U -q |
15 | 28 | echo "- Installing mmdet"
|
16 |
| -pip install mmdet==2.13.0 --upgrade -q |
17 |
| -echo "icevision installation finished!" |
| 29 | +pip install mmdet==2.13.0 -U -q |
| 30 | +echo "icevision installation finished" |
18 | 31 |
|
19 | 32 | # restart notebook
|
20 | 33 | echo "Restarting runtime!"
|
|
0 commit comments