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版本。类似于上一篇帖子