tensorflow-gpu 安装

废话少说直接来重点!!!!

1.安装Anaconda,一般在官网下最新的就行,没啥大的要求

接着是  Anaconda的配置


   一般为了独立,往往需要新建一个虚拟环境来配置tensorflow,先建立一个虚拟环境在开始菜单找到Anaconda Prompt(anaconda 3) 右键管理员运行

重点要装python3.7版本,因为我的显卡是1650ti,cuda是11.0版本,所以tensorflow版本要求较高

conda create -n tfgpu python==3.7 anaconda

2.这样就建立了tensorflow的虚拟环境

  • 切换到新建的虚拟环境
conda activate tfgpu

在这里先要配置好cuda和cdnn才行,先做之后步骤....

3.安装cuda cdnn

这里借鉴别人的博客安装,注意版本,非常重要!!!!

(7条消息) tensorflow安装GPU版本(含CUDA和CUDNN的安装)【windows】,超详细个人总结!_行野的博客-CSDN博客

作为自己版本的cuda11.0 cudnn选择v8.0.5  tensorflow-gpu 2.4.0

这是自己显卡支持的版本

 

这是在官网下载cdnn图,自己要注册账号,这步骤也很简单

4.重新接着2步骤来

  现在再次进入tensorflowenv虚拟环境

 

  • 下载tensorflow-gpu  这里网上有很多安装例如

pip install tensorflow==2.4.0
  •  

    conda install tensorflow==2.4.0
  • 一般加入豆瓣镜像下的比较快自己是用下面这行命令安装成功的,这里只要可以安装成功就行,自己在这一步卡了很久,也看了很多方法

  •  

  • 最后查看是否成功

  • 最后输出true就成功了

最后重点主要是版本对应和命令行安装tensorflow-gpu比较难,其他还好

 



 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值