【linux】安装cuda+cudnn
检查Driver
Note:
- 一台机器只能对应一个nvidia driver，而一个nvidia driver可以对应多个cuda。只要在自己的路径下把cuda指定成特定版本就行。
- 尽量不要使用/usr/local/下面的默认cuda/，防止因为ln对象会发生变化而造成影响。
- nvidia driver版本 决定了 可用的cuda范围，进而决定了 可用的tensorflow-gpu版本。所以，每次上一台新机器前，首先确定nvidia driver版本。
查看nvidia driver版本：
cat /proc/driver/nvidia/versionTable 1. CUDA Toolkit and Compatible Driver Versions：
屏幕快照 2019-09-18 上午11.35.08.png
安装Cuda
查看本机操作系统：
cat /etc/issue前往英伟达官网，下载对应的cuda版本：
屏幕快照 2019-09-18 下午2.18.11.png
Note：
- 以下以cuda9.0为例。
安装cuda：
sudo sh cuda_9.0.176_384.81_linux.runNote:
- 勿装driver！否则会安装失败！
屏幕快照 2019-08-12 下午10.07.12.png
写入对应路径：
vim ~/.bashrc
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
source ~/.bashrc查看cuda版本，验证是否安装成功：
nvcc -V安装Cudnn
下载对应的cudnn安装包。
Note：
- 以下以cuda9.0、cudnn7.6.2为例。
解压：
mkdir cudnn
tar -xvzf cudnn-9.0-linux-x64-v7.6.2.24.tgz -C cudnn/复制文件：
sudo cp cudnn/cuda/lib64/lib* /usr/local/cuda-9.0/lib64/
sudo cp cudnn/cuda/include/cudnn.h /usr/local/cuda-9.0/include/设置软链接：
cd /usr/local/cuda-9.0/lib64
sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.6.2 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2020-03-15 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录