`由于3090只能安装cuda11以上的版本,与作者描述的步骤有些不同,所以总结一下问题,避免看到的小伙伴走弯路
`提示:与作者提供的顺序大致相同,但需要更改cuda版本
conda create --name pytracking python=3.7
conda activate pytracking
2.
conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
*如果添加清华镜像,去掉-c python速度更快,且不易失败。(-c python指定下载路径为pytorch官网)
另外,由于torch.rfft不能在torch1.7以上的版本运行,下面会介绍这个错误。
3.
conda install matplotlib pandas tqdm
pip install opencv-python visdom tb-nightly scikit-image tikzplotlib gdown
4.
conda install cython
pip install pycocotools
pip install lvis
5.
sudo apt-get install ninja-build
6.
`pip install spatial-correlation-sampler
7.
sudo apt-get install libturbojpeg
pip install jpeg4py
问题1./bin/sh: 1: /usr/local/cuda:/bin/nvcc: not found
解决:在终端运行:export CUDA_HOME=/usr/local/cuda-11.0,然后再次运行代码。
问题2.nvcc fatal : Unsupported gpu architecture ‘compute_86’
解决:在终端运行:export TORCH_CUDA_ARCH_LIST=“8.0”
*提示:有的解决方法为export TORCH_CUDA_ARCH_LIST=“7.5”,这个是不可以解决问题的,而且会引来新的错误:CUDA Error: no kernel image is available for execution on device。
问题3.module ‘torch’ has no attribute ‘rfft’
解决:安装torch1.7.1,上面提到了。torch1.8及以上都没有rfft。