ubc-vision/COTR

Conda env

Opened this issue · 0 comments

I found the following command can install the basic env for inference:

conda create -n cotr_env python=3.7
conda activate cotr_env
conda install scipy=1.2.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch
conda install opencv=3.4.2
conda install scikit-image jupyter pytables matplotlib imageio tqdm
pip install tensorboardX
pip install easydict
pip install kornia==0.5.3
pip install --upgrade vispy