出问题概率最低,最稳妥的用Anaconda安装TensorFlow-gpu版本方法

文章提供了使用conda安装tensorflow-gpu的简便方法,强调无需额外下载cuda和cuDNN,conda会自动处理依赖。只需确保gpu驱动和cuda版本兼容,如3060支持cuda11.3。通过condainstalltensorflow-gpu=2.6.0命令进行安装,并使用condasearch命令检查可用版本。
摘要由CSDN通过智能技术生成

目录

一. 直接上安装方法

二. 版本适配查看


一. 直接上安装方法

conda install tensorflow-gpu=2.6.0

打开Anaconda Prompt之后,激活你的环境,输入这一行就够了,不只是tensorflow-gpu版本,他会给你把所有需要的cudatoolkit包,cuDNN包在conda里一起安装(2.6.0这个版本需要看情况改,见下文)

  接下来只需要等待就能安好

重点,安装TensorFlow-gpu版本,只需要你提前安一个显卡驱动即可,不用换镜像源,不用提前去英伟达cuda官网,不用去cuDNN官网下一堆再安一堆东西(安装一整个cuda还占电脑更多硬盘,conda自动只会安装TensorFlow需要的cudatoolkit)

网上一堆教程里教你下载cuda,cuDNN对初学者来说是多此一举!!!我试了这个试了那个都不行还有各种报错,还得找对应的各种版本,太痛苦了,最后发现最好的办法就是创建新环境(专门用TensorFlow的)再直接Anaconda Prompt直接安装,没有任何多余的操作

二. 版本适配查看

2.6.0这个版本需要你的GPU支持cuda11.3,cuDNN8.2,主要是得支持这个cuda版本(2023/2/24笔记本的3060最高支持到cuda12.0,向下兼容,所以11.3能用,能安的最新的就是TensorFlow-gpu的2.6.0版本),可以用Nvidia驱动控制面板的左下角看系统信息,先点1打开系统信息面板,2这行显示这个驱动支持的最高的cuda版本

用这两个指令

conda search tensorflow-gpu
conda search cudatoolkit

查看需要安装时conda支持直接安装的tensorflow-gpu版本和能直接自动安的cudatoolkit版本

下图是范例,例如你的GPU支持的最高cuda版本高于11.2,直接安装TensorFlow2.6.0,如最高版本支持到cuda11.0,安装TensorFlow2.4.0就可以了

图片来源((3条消息) 2021最新:TensorFlow各个GPU版本CUDA和cuDNN对应版本整理(最简洁)_Kakaluotuo的博客-CSDN博客_cuda12.0对应cudnn

  • 3
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值