TensorFlow_gpu版本 python库安装流程（包括驱动，cuda，cudnn）

最新推荐文章于 2022-11-25 21:47:31 发布

luffytom

最新推荐文章于 2022-11-25 21:47:31 发布

阅读量762

点赞数 1

分类专栏：深度学习文章标签： tensorflow gpu 库

本文链接：https://blog.csdn.net/luffytom/article/details/78448230

版权

本文详细介绍了在Ubuntu系统中安装NVIDIA驱动、CUDA、cuDNN以及TensorFlow GPU版的步骤，包括禁止nouveau驱动、设置环境变量等关键环节，旨在帮助读者成功配置GPU加速的TensorFlow环境。

摘要由CSDN通过智能技术生成

安装NVIDIA显卡驱动

由于Ubuntu系统自带的驱动更新是无法更新的显卡的最新驱动的，而且在更新过程中容易出现问题，所以我们本次选择了本地安装驱动。

下载驱动：

http://www.geforce.cn/drivers，在NVIDIA网站上选择对应的网卡型号和驱动版本进行下载。

选择对应显卡

选择了之后在下方会出现该显卡的可下载版本驱动：

这里写图片描述

禁止Ubuntu集成的nouveau驱动

Ubuntu系统集成的显卡驱动程序是nouveau，它是第三方为NVIDIA开发的开源驱动，我们需要先将其屏蔽才能安装NVIDIA官方驱动。
将驱动添加到黑名单blacklist.conf中，但是由于该文件的属性不允许修改。所以需要先修改文件属性。
- 修改黑名单属性： $sudo chmod 666 /etc/modprobe.d/blacklist.conf
- 修改名单：

$sudo vim /etc/modprobe.d/blacklist.conf

### 在打开了文件以后在文件最后添加几行：
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb