Ubuntu18.04 tensorflow-gpu环境配置

本文档详细介绍了如何在Ubuntu18.04上配置tensorflow-gpu环境,包括安装Nvidia驱动、CUDA、CuDnn以及tensorflow-gpu的步骤。在安装过程中,特别提到了禁用nouveau的必要性,但建议不必禁用X服务。最后,还给出了安装完成后验证驱动和环境的方法,并提供了强化学习环境配置的参考资料。

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


TensorFlow 官方文档: TensorFlow 2.0.0 安装

准备安装文件

  1. 下载Nvidia显卡驱动:NVIDIA驱动版本选择
  2. 下载CUDA:CUDA版本选择,选择要下载的版本,在‘select target platform’中依次选择Linux, X86_64, Ubuntu, 18.04, deb(local). 我的安装经验发现,runfile(local)安装过程看起来步骤少一些,但实际上一直报错无法安装。
  3. 下载与CUDA对应的CuDnn版本:Cudnn版本选择

安装Nvidia驱动

  1. 一般下载的安装文件是放在~/Downloads路径下, 通过cd ~/Downloads 到达安装文件所在的路径.

  2. 安装依赖项:sudo apt-get install gcc g++ make

  3. 禁止nouveau
    ➢ 打开要编辑的文件:sudo gedit /etc/modprobe.d/blacklist.conf
    ➢ 在文件末尾加入:
    blacklis

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值