这个问题困扰了我很久。今天终于解决了,记录一下。
造成这个结果的原因是cuda编译版本与runtime版本不符。
安装的是10.0版本,也就是/usr/local/cuda-10.0
(默认安装目录)。
nvcc -V
的结果是7.5。
apt list|grep nvidia-cuda-tookit
显示版本为7.5,找到原因了。
卸载掉
sudo apt autoremove nvidia-cuda-toolkit
把安装版本加入环境变量
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
source ~/.bashrc
此时在执行nvcc -V
显示正确版本。