GPU版本tensorflow安装

由于tensorflow默认安装都是CPU版本,之前一直安装gpu版本出现很多问题,这里终于找到一种安装方法,以及自己电脑对应的版本。供大家参考。其中找到的比较优质的博文参考如下
(1)[精简]2023年最新windows安装GPU版本tensorflow
(2)CUDA12.2版本下tensorflow-gpu安装教程
(3)机器/深度学习笔记:tensorflow(gpu版本)在win10系统上的安装方法

安装CUDA

在NVIDIA官网上安装cuda toolkit,这里安装的是最新的12.3版本的。
安装过程直接按照推荐设置,安装精简版本,未进行修改。安装完成后在下面路径下出现12.3的文件夹
在这里插入图片描述

安装CUDNN

也是NVIDIA官网上安装,然后解压到CUDA所在的文件夹下。在我的电脑-高级电脑设置-环境变量-PATH编辑,增加以下两个路径
在这里插入图片描述
在这里插入图片描述

在anconda环境中安装tensorflow-gpu

即使cuda和cudnn安装正确,稍不注意,还是可能由于版本问题而导致gpu版本的tensorflow无法使用。按照下面流程,再加上不断尝试,最终总结出了两个版本的内容供大家选择。当然,大家也可以根据电脑配置自行选择,选择依据主要参考官网上的对应版本。
在这里插入图片描述

查看cuda/cudnn版本

conda search cuda
conda search cudnn

出现的内容如下
在这里插入图片描述

不同版本Python安装

Python3.8版本

conda create -n tensorflowgpu python=3.8.18
conda search cuda
conda search cudnn
conda install cudatoolkit=10.1.243
conda install cudnn=7.6.5
pip install tensorflow-gpu==2.7.0
pip install matplotlib==3.3.4

中间还遇到一些错误,上面的cuda包似乎不太对,,,可能我的电脑有高版本的CUDA和cudnn,反正最终的包版本如下所示
在这里插入图片描述

Python3.6版本

在版本二中,只能安装2.2.0版本的tensorflow
安装的命令如下

conda create -n tensorflownew python=3.6.13
conda search cuda
conda search cudnn
conda install cudatoolkit=10.1.243
conda install cudnn=7.6.5
pip install tensorflow-gpu==2.2.0
pip install matplotlib==3.3.4

安装的包版本如下,把matplotlib特意标注出来,因为之前遇到过高版本的matplotlib对应的numpy版本与tensorflow版本要求不一致。这里采用的是低版本的3.3.4版本基本上都可以。
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值