Ubuntu 安装 Anaconda 和 tensorflow-gpu 的关键步骤

本文提供了在Ubuntu 20.04上安装Anaconda并配置TensorFlow-GPU的关键步骤,强调了系统、驱动、TensorFlow版本间的匹配。避免了直接安装CUDA和CuDNN,而是通过Anaconda自动安装。遇到的问题包括安装错误导致的训练速度慢和精度低,解决办法是按照正确步骤重新安装。最后,给出了CPU和GPU运行性能对比。

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

经过一些折腾,中间重装过系统、碰到一些奇怪的问题,最后环境搭建成功。

网上已经有很多内容描述细节,因此本文只把关键步骤写下来做参考。

<1> 正确步骤

关键的点:

  • 操作系统、GPU驱动、tensorflow版本、CUDA版本、CuDNN版本需要匹配上。这步看似废话,但是至关重要,否则就可能会碰到各类“难缠”的奇怪问题
  • 如果使用 Anaconda 安装tensorflow 开发环境,那么操作系统安装后,只需要在操作系统上直接安装 GPU 驱动,而不用安装CUDA 和 CuDNN 驱动,因为 Anaconda 安装tensorflow 时会自己在 Conda Env 里安装 Cuda 和 CuDNN 驱动

成功安装的步骤:

  • 安装操作系统:ubuntu 20.04
  • 在操作系统上直接安装: nvidia gpu驱动 510.73.05(GPU是rtx 3060 laptop)
  • 安装Anaconda
  • 在Anaconda上:安装tensorflow 开发环境,安装 tensorflow-gpu 2.8,cuda 11.6.0(安装 tf-gpu时自动顺带安装),cudnn 8.4.0.27(安装 tf-gpu时自动顺带安装)
  • 安装 pycharm,然后在项目中绑定 conda env,之后便可以在里面开发运行调试代码

安装步骤中 conda 相关的几个命令:

conda create -n TF280-GPU python=3.8.0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值