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

最新推荐文章于 2023-11-07 21:10:56 发布

原创

最新推荐文章于 2023-11-07 21:10:56 发布 · 608 阅读

0 ·

CC 4.0 BY-SA版权

本文详细介绍了如何在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，安装驱动

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