亲测不用先安装显卡,可以直接安装CUDA,CUDA内集成了最新显卡驱动。从新安装了4次系统,很多一上来说协助原来的驱动,禁用显卡驱动,一大堆操作。很多都是操蛋。
显卡是gtx1080,驱动版本是418.67,CUDA版本是10.1,cuDNN版本是7.6,系统是ubuntu16.04
安装上面的操作步骤,1,2,3,4来
重点是第2步,当运行完第一步sudo dpkg -i cuda-repo-ubuntu1604-10-1-local-10.1.168-418.67_1.0-1_amd64.deb以后。
在终端会显示第二步sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub 全部完整命令。
完成安装以后,安装cuDNN
我没有配置环境变了,全部安装完以后,运行DL4J,没有任何问题
点击:cuDNN Library for Linux 下载,解压
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
查看cudnn版本
在终端输入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
显示如下表示成功