基于ubuntu16.04操作系统安装tensorlfow-gpu版(GPU板卡：nvidia tesla P100)

Ubuntu16.04安装TensorFlow-GPU与NVIDIA Tesla P100配置指南

最新推荐文章于 2025-10-11 09:36:00 发布

原创

最新推荐文章于 2025-10-11 09:36:00 发布 · 1.8k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow-gpu #tesla P100 #cuda #cudnn #ubuntu16.04

本文详细介绍了如何在Ubuntu16.04操作系统上安装TensorFlow-GPU版，针对配备NVIDIA Tesla P100的GPU板卡，包括CUDA 9.0、cuDNN 7.2的安装步骤，以及驱动和环境变量的配置。通过验证脚本确保安装成功，并提供了后续使用TensorFlow-GPU的指引。

一、Tensorflow-gpu安装需要
根据tensorflow-gpu版本的软件要求：
【注意】NVIDIA GPU drivers 需要与CUDA对应
CUDA 9.0
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux
NVIDIA GPU DRIVERS >= 384.x
https://www.nvidia.cn/Download/driverResults.aspx/139366/cn
cuDNN >= 7.2
https://developer.nvidia.com/rdp/cudnn-download

二、TESLA P100驱动安装
1、首先根据自己GPU的型号到NVIDIA官网去下载相应的驱动程序：
不是最新最好，而是要版本对应才最好；
本团队使用的GPU：TESLA P100
根据自己电脑的操作系统；GPU的型号；CUDA版本；下载对应的驱动文件；

2、   下载好之后是一个.run文件；
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
安装驱动需要注意的几点:
关闭图形界面：
(1)   sudo systemctl set-default multi-user.target (启动时默认多用户文本界面)
sudo systemctl set-default graphical.target (启动时默认图形界面)
(2)   或者也可以使用，关闭：sudo service lightdm stop
恢复：sudo service lightdm stop
(3)   关闭显卡驱动：
创建：/etc/modprobe.d/blacklist-nouveau.conf
编辑：blacklist nouveau