- Andconda
- install Cuda11.1 and cudnn (requested by imaginaire) If you are not root user, you can install cuda in you '~' path
- make sure the 'nvcc -V' & 'cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2' have a right output
- apt-get update & apt-get install cmake (if you are not in docker content, you should add 'sudo before every apt-get')
- conda activate your-env-name (make sure python>3.8)
- pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111 or https://mirror.sjtu.edu.cn/pytorch-wheels/cu111/torch_stable.html
- test 'torch.cuda.is_available()'
- bash scripts/install