可能因为自己GPU太新,cuda版本不支持
我的cuda版本是9.1,所以我又安装了一个cuda10.0版本
1. 下载cuda10.0版本
官网地址:链接: https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal.
选择 Linux -> x86_64 -> Ubuntu -> 18.04 -> runfile(local) (根据自己电脑版本选择)
2. 安装cuda10.0版本
我下载的是cuda_10.0.130_410.48_linux.runfile。
下载好后进入到cuda_10.0.130_410.48_linux.run 所在的目录,执行下面两条命令安装:
sudo chmod +x cuda_10.0.130_410.48_linux.run # 为 cuda添加可执行权限
sudo ./cuda_10.0.130_410.48_linux.run # 安装
接下来,一直长按enter,会出现以下界面,一步一步照着做就行
Do you accept the previously read EULA?
accept/decline/quit: accept # 接受协议
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n # 是否安装显卡驱动包,由于已经安装显卡驱动,选择n
Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: y # 是否安装工具包,选择y
Enter Toolkit Location
[ default is /usr/local/cuda-10.0 ]: # 工具包安装地址,默认回车即可
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: n # 如果之前安装过另一个版本的cuda,除非你确定想要用这个新版本的cuda,否则这里就建议选n,因为指定该链接后会将cuda指向这个新的版本
Install the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: y # 安装样例
Enter CUDA Samples Location
[ default is /home/jiaoda ]: # 样例安装地址默认即可
Installing the CUDA Toolkit in /usr/local/cuda-10.0 ...
Missing recommended library: libGLU.so
Missing recommended library: libXmu.so
Missing recommended library: libGL.so
Installing the CUDA Samples in /home/jiaoda ...
Copying samples to /home/jiaoda/NVIDIA_CUDA-10.0_Samples now...
Finished copying samples.
***安装完成***
3. cuda环境设置
终端首先输入:
gedit ~/.bashrc
在打开的 .bashrc末尾添加环境变量:
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
注意上面的路径都是指向/usr/local/cuda 软连接,并没有写死指向某一个cuda版本,后面切换时不用改路径,只改软连接指向就可以。
保存关闭后,执行下面命令,使之立即生效
source ~/.bashrc
4. cuda版本切换
到/usr/local/目录下打开终端
sudo rm -rf cuda #删除之前的软链接(原本指向9.1)
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda #生成新的软链接,指向10.0
以下命令均可以查看当前cuda软链接指向的哪个cuda版本,选哪条命令都可
nvcc --version
nvcc -V
如果你想使用之前的cuda-9.1版本时,只需要删除上面的软链接,然后重新建立指向cuda-9.1版本的软链接即可
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
参考链接 https://blog.csdn.net/qq_45057749/article/details/107320354.