TensorFlow GPU版本安装(Windows)

 TensorFlow API Versions版本icon-default.png?t=O83Ahttps://tensorflow.google.cn/versions

 

平时一般用使用pytorch多一点,TensorFlow也是比较流行的深度学习框架,想来也是研究一下。目前最新的TensorFlow版本为2.17.0,但是直接下载,GPU不可用;遂来整理一下。

其他帖子提出,从tensorflow 2.11.0版本开始,在windows上不再支持GPU;但是我换个2.10.0版本下载,貌似也没有效果,应该是cuda和cudnn版本的问题。(之前使用pytorch时下载了高版本)

在1.x版本之前的版本CPU和GPU是分开的,我们在Windows系统下使用TensorFlow进行GPU加速就有两种形式。

1、可以尝试安装1.15.0的tensorflow-gpu版本,一般情况下现在流行的框架也是兼容的。

2、安装tensorflow-gpu 2.10.0及以前的版本

第一种比较简单,可以直接在虚拟环境中一键安装;后一种就是需要搭配对应的cuda版本和cudnn版本。类似于上一篇帖子

pytorch-cuda(GPUtorch详细教程)_pip install torch== cu121-优快云博客icon-default.png?t=O83Ahttps://blog.youkuaiyun.com/bookye_/article/d

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值