先说明主机安装好Ubuntu18.04后什么都不要安装,包括显卡驱动,cuda里面有带安装显卡驱动的。直接下载cuda10.0和对应版本的cudnn,主要因为现阶段最高版本TensorFlow貌似支持cuda10.0,不支持cuda10.1应为我安装cuda10.1后运行TensorFlow出错。所以后面又卸载cuda10.1,安装cuda10.0.
一、安装cuda
1.下载
上官网下载(cuda官网)
查看下载以往版本,进去后选择cuda10.0下载
然后按照版本选择下载run包
2.安装
按照说明,打开终端运行:
sudo sh cuda_10.0.130_410.48_linux.run
终端出现安装界面,出现协议,按住回车键一直跑(如果你想看一下也可以一条一条往下看,嘻嘻……)
因为我之前已经安装过显卡了,所以第一项显卡那里我选择了n,其他都是y,如果你的ubuntu是干干净净的,就直接全部选择y。接下来就是等待安装成功!之后路径写入环境变量也就是终端输入(要在根目录下):
sudo gedit .bashrc
如果不在根目录则输入:
sudo gedit ~.bashrc
在写入路径:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并退出,重启.bashrc文件:
source ~.bashrc
3.测试
在终端输入:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
出现如图,表示安装成功:
二、安装cudnn
1.下载
下载对应版本的cudnn(cudnn下载需要登录网站,可以用邮箱注册,然后下载压缩文件,如图所示):
2.安装
直接用Ubuntu右键解压到当前文件,后进入文件,打开终端运行:
1. sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
2. sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
3. sudo chmod a+r /usr/local/cuda/include/cudnn.h
4. sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
在终端查看CUDNN版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
至此,cuda和cudnn全部安装成功!
中间参照许多大牛的博客,在此表示非常感谢!!!