因为nvidia 驱动为390.67 与cuda 9.2 不匹配,无法使用,所以对cuda版本降级:
过程:
卸载cuda 9.2(可删可不删,不删会保留cuda-9.2和cuda-9.1两个文件夹)
a@a-6108GC:~$ cd /usr/local/cuda-9.2/bin/
a@a-6108GC:/usr/local/cuda-9.2/bin$ sudo ./uninstall_cuda_9.2.pl
我选择删除干净。
安装CUDA 9.1
该过程可参考网上信息了,先从官网下载cuda历史版本cuda-9.1,网站为:cuda历史版本。
我下载了9.1.85XXX.run,以及补丁,并将两个文件拷贝到home目录下。
以下操作都在命令行界面操作,比如按下Ctrl+alt+F1进入命令行
首先停止lightdm:
sudo service lightdm stop
然后,给两个文件权限并运行cuda驱动
sudo chmod 777 cuda_9.1.85*
./cuda_9.1.85_387.26_linux.run
第一个问题“是否安装nvidia 驱动”选'no',
然后科研到Samples目录下,先make编译,然后到bin下找到deviceQuery并执行。
有结果则表明成功安装cuda驱动。
安装cuDNN
cuDNN是NVIDIA专为Deep Learning应用开发的支持库。
到这里下载
下载:cudnn-9.1-linux-x64-v7.tgz
。
将其传到/usr/local
目录下,然后解压即可:
sudo tar -zxvf cudnn-9.1-linux-x64-v7.tgz
这样就成功安装了CUDA 9.1这一套,但是驱动仍然用的高版本驱动390.25,不过应该没关系的吧。
添加环境变量
vim ~/.bashrc
添加:
export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64:/usr/local/cuda-9.1/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-9.1
export PATH=$CUDA_HOME/bin:$PATH
source .bashrc