TensorFlow 2.X GPU版本的安装

1.打开 anaconda Prompt(Anaconda3)

输入

pip install tensorflow==3.0

得到结果

ERROR: Could not find a version that satisfies the requirement tensorflow==3.0 (from versions: 2.5.0, 2.5.1, 2.5.2, 2.5.3, 2.6.0rc0, 2.6.0rc1, 2.6.0rc2, 2.6.0, 2.6.1, 2.6.2, 2.6.3, 2.6.4, 2.6.5, 2.7.0rc0, 2.7.0rc1, 2.7.0, 2.7.1, 2.7.2, 2.7.3, 2.8.0rc0, 2.8.0rc1, 2.8.0, 2.8.1, 2.8.2, 2.9.0rc0, 2.9.0rc1, 2.9.0rc2, 2.9.0, 2.9.1, 2.10.0rc0, 2.10.0rc1, 2.10.0rc2)

 可以看到现在tensorflow的版本号

如果想下载cpu版本，使用命令

pip install 版本号

如果想下载CPU版本需要先下载CUDA，下载网址如下：

NVIDIA Developer

这里我的CUDA的版本是11.7

然后再下载cuDNN，下载网址如下

cuDNN Archive | NVIDIA Developer

需要选择cuDNN和cuda相互对应的版本

需要注册账户后才可下载

需要将其解压到CUDA的目录中，其中我的目录是

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7

将CUDA添加进入环境变量即可；

此时我们在anaconda prompt中使用命令

pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

 然后使用

pip list

检查tensoflow版本号

 这是我的版本号

要是各位小伙伴对tensorflow2.x的安装还有什么问题，欢迎在下方评论区留言或直接私信。

 参考：TensorFlow-GPU版本安装（CUDA11.1）_biu piu biu piu的博客-CSDN博客_cuda11.1 tensorflow