Ubuntu18:装GPU TensorFlow

原创

于 2018-05-10 22:08:03 发布 · 7.1k 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu18 #tensorflow #gpu #cuda #cuDNN

本文详细介绍了如何在Ubuntu18.04系统上安装CUDA9.0、cuDNN7.0.5以及GPU版本的TensorFlow。包括禁用nouveau驱动、安装Nvidia驱动、配置CUDA环境变量以及测试CUDA和cuDNN的步骤，最后验证TensorFlow成功运行在GPU上。

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

环境：ubuntu18.04LTS，python3.6.5
目标环境：cuda9.0 cuDNN7.0.5_for_cuda9.0 tensorflow任意

2018.10.24更新

ubuntu18装nvidia驱动可以在设置里面切换了，所以可以直接从cuda装起，很简单。

更新end

2018.09.17更新

突然发现一条命令就可以装nvidia驱动

ubuntu-drivers autoinstall

更新end

2018.05.10原文如下：
安装依赖：

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libfreeimage3 libfreeimage-dev

卸载原有N卡驱动（如果有的话）：

# 建议run文件卸载，即你之前下载的Nvidia驱动run文件
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall
# 不建议采取这种
sudo apt-get remove --purge nvidia*

禁用nouveau驱动（十分重要）：

sudo gedit /etc/modprobe.d/blacklist.conf

文本最后添加：

blacklist nouveau
options nouveau modeset=0

命令行使生效：

sudo update-initramfs -u

重

最低0.47元/天解锁文章

200万优质内容无限畅学