Ubuntu18:装GPU TensorFlow

ShellDawn

于 2018-05-10 22:08:03 发布

阅读量7.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： ubuntu 文章标签： ubuntu18 tensorflow gpu cuda cuDNN

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/80274534

本文详细介绍了如何在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万优质内容无限畅学