Skip to content

Commit

Permalink
Added script for icevision inference installation (#893)
Browse files Browse the repository at this point in the history
* added script for icevision inference installation

* install torchvision and torchtext first
  • Loading branch information
ai-fast-track authored and lgvaz committed Aug 5, 2021
1 parent 94d7e0a commit e7eb73f
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions install_icevision_inference.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
platform="${1}"
case ${platform} in
colab)
echo "- Installing icevision inference in Colab"
echo "Uninstalling some dependencies to prevent errors"
pip uninstall fastai -y
pip uninstall torchvision -y
pip uninstall torchtext -y

echo "- Updating torchvision"
pip install torchvision==0.9.0 -U -q

echo "- Updating torchtext"
pip install torchtext==0.9.0 -U -q

echo "Installing some dependencies to prevent errors"
pip install PyYAML>=5.1 -U -q
pip install datascience -U -q
pip install tensorflow==2.4.0 -U -q
pip install google-colab -U -q


echo "- Installing icevision and icedata"
pip install git+git://github.com/airctic/icevision.git#egg=icevision[inference] -U -q
pip install icedata -U -q
echo "icevision installation finished!"
;;
*) # Installing only icevision and icedata
echo "- Installing icevision inference"
pip install git+git://github.com/airctic/icevision.git#egg=icevision[inference] -U -q
pip install icedata -U -q
echo "icevision installation finished!"
;;
esac

0 comments on commit e7eb73f

Please sign in to comment.