Ubuntu18.04安装CUDA/CUDNN
2020-05-13 本文已影响0人
AntiGravity
硬件:RTX2080
1. 安装驱动
sudo apt-get install nvidia-driver-440 # 440是驱动版本号,输到前一位按tab后选择最新
测试:
nvidia-msi # 显示驱动版本440和驱动的CUDA版本10.2(和运行CUDA不同)
2. 安装CUDA
目前,为适配TensorFlow1.15建议使用CUDA 10.0版本。
TensorFlow各版本对CPU GPU版本的需求:官网
有一种说法,安装1.15的tensorflow-gpu后,自动安装相应的cuda和cudnn,不过我没有尝试过。
pip3 install tensorflow-gpu==1.15
3. 安装CUDNN
官网下载runtime/dev/doc三个版本的deb文件,依次安装。
官网说明
- 安装后查看版本:
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 用sample验证CUDNN(需要之前安装dev和doc两个deb)
修改cudnn.h的include "..."为include <...>,否则下面编译会报错
cd /usr/src/cudnn_samples_v7/mnistCUDNN # 如果没有操作权限就cp到可操作的位置
make clean && make
./minstCUDNN