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

修改cudnn.h的include "..."为include <...>,否则下面编译会报错

cd  /usr/src/cudnn_samples_v7/mnistCUDNN # 如果没有操作权限就cp到可操作的位置
make clean && make
./minstCUDNN
上一篇下一篇

猜你喜欢

热点阅读