首先得安装cuda和torch,版本要一致
先用命令nvcc -V查看是否安装了cuda以及cuda版本
若是没安装cuda
再用nvidia-smi查看显卡驱动,去NVIDIA官网查看匹配的cuda版本
在CUDA Toolkit Archive | NVIDIA Developer下载对应的cuda版本,根据命令安装完后
vim ~/.bashrc
# 单击i以编辑,在文件中添加对应版本的内容
CPATH=/usr/local/cuda-11.8/targets/x86_64-linux/include:$CPATH
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/targets/x86_64-linux/lib:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.8
单击Esc,输入:wq以退出
最后要记得source ~/.bashrc
最后用命令nvcc -V查看
然后去github将Pytorch-Correlation-extension拷贝下来将其解压
在Pytorch-Correlation-extension目录下打开终端
在终端再次source ~/.bashrc一遍(我就是出错在这个地方,导致安了好几遍都不成功)
conda activate (自己的环境)
python setup.py install
最后不出意外就安装成功了