文章目录
ubuntu 16.04 安装 CUDA9.0 和 cuDNN7
1. 英伟达驱动安装
我的是笔记本 GTX965M 显卡。
1.1. 方法一、ubuntu 系统自动安装(我只用过这个方法)
1.2. 方法二、命令行
sudo apt-get install nvidia-384 nvidia-modprobe
1.3. 查看是否安装成功
nvidia-smi
2. CUDA
2.1. 下载 CUDA9.0
下载中链接:https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
2.2. 安装 CUDA9.0
sudo sh cuda_9.0.176_384.81_linux.run
sudo bash -c "echo /usr/local/cuda/lib64/ > /etc/ld.so.conf.d/cuda.conf"
sudo ldconfig
sudo vim /etc/environment
添加(:/usr/local/cuda/bin)进去:add :/usr/local/cuda/bin (including the ":") at the end of the PATH="/blah:/blah/blah" string (inside the quotes)
reboot
2.3 验证是否安装成功(查看版本)
cat /usr/local/cuda/version.txt
3. cuDNN (推荐)
3.1 下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive
3.2 解压 下载的 .tgz 文件
将下载的问价复制到 /opt 目录下, 解压 下载的 .tgz 文件
tar zxvf 文件名.tgz
3.3 安装
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*
3. cuDNN(不推荐)
3.1. 下载 cuDNN
需要注册账号,免费下载,三个文件都要下载: https://developer.nvidia.com/rdp/cudnn-download
3.2. 安装方法
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda9.0_amd64.deb
# put the following line in the end or your .bashrc file
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64"
source ~/.bashrc
3.3. 查看是否安装成功
cp -r /usr/src/cudnn_samples_v7/ ~
cd ~/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
# you should see "Test passed"
3.4. 查看 cuDNN 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
4. 参考的链接(非常有用):
https://medium.com/repro-repo/install-cuda-and-cudnn-for-tensorflow-gpu-on-ubuntu-79306e4ac04e
https://blog.csdn.net/wanzhen4330/article/details/81699769#cudnn%E7%9A%84%E5%AE%89%E8%A3%85