The code was tested with Anaconda Python 3.10.13, CUDA 12.1, and PyTorch 2.1.2. After installing Anaconda:
-
[Optional but highly recommended] create a new conda environment.
conda create -n svanet python=3.10.13
And activate the environment.
conda activate svanet
-
Install PyTorch:
conda install pytorch=2.1.2 torchvision=0.16.2 pytorch-cuda=12.1 -c pytorch -c nvidia
-
Clone this repo:
git clone https://github.com/anthonyweidai/SvANet.git
-
Install the requirements
cd $SvANet_ROOT pip install -r requirements.txt