Having anaconda installed on a ubuntu 20.04 box:
Create a virtual environment, specifying the disk:
conda create --prefix /mnt/stockage/Developp/EnvPLFlash
and activate the env with:
conda activate /mnt/stockage/Developp/EnvPLFlash
Then install the libs starting with pytorch with cuda support:
To have pytorch 1.8 with cuda support:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-lts
then
pip install icedata
pip install lightning-flash
pip install notebook
pip install voila
Without forgetting to install lightning-flash[image] to get the instance segmentation algorithms
pip install 'icevision' 'lightning-flash[image]'
The installation can be checked running the following notebook:
