ubuntu18.04安装tensorflow-gpu 驱动gtx1060+CUDA Toolkit 9.0+ CUDNN 7.0+libcupti+配置+安装tensorflow-gpu+测试

最新推荐文章于 2020-11-21 15:09:30 发布

doublewei1

最新推荐文章于 2020-11-21 15:09:30 发布

阅读量607

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/doublewei1/article/details/85463873

本文详细介绍了如何在Ubuntu 18.04上安装TensorFlow-GPU，包括安装CUDA Toolkit 9.0、cuDNN 7.0、libcupti，以及配置环境变量和测试步骤。确保拥有root权限，并遵循文中提供的命令进行操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

具备条件：root权限进行操

修改root密码：

1.  $ sudo passwd    输入两次新密码

2.  $ su root       登陆 root账户

显卡驱动安装：

step .1：首先，检测你的NVIDIA图形卡和推荐的驱动程序的模型。执行命令：
$ ubuntu-drivers devices
输出结果为：

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001180sv00001458sd0000353Cbc03sc00i00
vendor : NVIDIA Corporation
model : GK104 [GeForce GTX 680]
driver : nvidia-304 - distro non-free
driver : nvidia-340 - distro non-free
driver : nvidia-384 - distro non-free recommended
driver : xserver-xorg-video-nouveau - distro free builtin

== cpu-microcode.py ==
driver : intel-microcode - distro free
从中可以看到，这里有一个设备是GTX 680 ，对应的驱动是NVIDIA -304,340,384 ，而推荐是安装384版本的驱动。

step.2，安装驱动

你可以选择，安装所有推荐的驱动，如下命令