ubc-vision / COTR

Code release for "COTR: Correspondence Transformer for Matching Across Images"(ICCV 2021)
Apache License 2.0
464 stars 58 forks source link

Conda env #25

Open jiangwei221 opened 2 years ago

jiangwei221 commented 2 years ago

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