Ubuntu18.04安装cuDNN和Tensorflow的正确姿势
本文章主要讲解 ubuntu 系统如何正确安装 cuDNN 和 Tensorflow。
需要提前准备:
一个安装好的Ubuntu 18.04
NVIDIA 驱动已经安装成功
CUDA已经安装成功
一、检查NVIDIA驱动是否安装成功
打开终端执行:
nvidia-smi
Wed Jul 10 11:49:26 2019
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.26 Driver Version: 430.26 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce RTX 208... Off | 00000000:42:00.0 On | N/A |
| 0% 42C P8 17W / 300W | 309MiB / 11011MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+------------------------------------------------------------------------