【cuda环境】python pytorch 报错CudaSetupArgument解决方法

@TOC
出现这个问题大概率是编译所使用的cuda版本和运行的cuda版本不一致
首先查看系统cuda版本（即编译所用的cuda版本）

nvcc -V

我用的pytorch+conda环境，使用conda list可以查看虚拟环境中cudatoolkit的版本，最开始我的系统cuda版本是9.0,cudatoolkit版本是10.2,因此出现了版本不一致的现象，所以出现了如题所示错误信息。后来我切换了系统的cuda版本，问题就解决了。
具体切换版本的方法简述：
echo $PATH 查看cuda PATH信息，添加cuda10.2的路径链接至/usr/local/cuda下，具体指令为

ln -s /usr/local/cuda10.2 /usr/local/cuda

然后修改系统PATH,具体操作如下

vim ~/.bashrc

在最后添加代码

export PATH=/usr/local/cuda:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

按esc键，输入:wq，回车键退出，然后在命令行输入

source ~/.bashrc

更新PATH信息
现在命令行输入

nvcc -V

可以查看切换后的cuda版本