TensorFlow API Versions版本
https://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版本。类似于上一篇帖子
GPU
| 版本 | Python 版本 | 编译器 | 构建工具 | cuDNN | CUDA |
|---|---|---|---|---|---|
| tensorflow_gpu-2.10.0 | 3.7-3.10 | MSVC 2019 | Bazel 5.1.1 | 8 |

5万+

被折叠的 条评论
为什么被折叠?



