美文网首页
Ubuntu18.04安装CUDA/CUDNN

Ubuntu18.04安装CUDA/CUDNN

作者: AntiGravity | 来源:发表于2020-05-13 00:14 被阅读0次

硬件: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

相关文章

网友评论

      本文标题:Ubuntu18.04安装CUDA/CUDNN

      本文链接:https://www.haomeiwen.com/subject/ekywnhtx.html