Anaconda安装Tensorflow-GPU

1.0 安装前的准备

第一步：查看显卡支持的CUDA版本

如图我的显卡支持的最高版本为11.6

第二步：打开链接拉到最后 自行选取CUDA和cuDNN版本

如下图

对于我来说 我可以安装的CUDA是11.6以下的所有版本
我选择了10.1CUDA，则对应的7.6cuDNN以及2.3tensorflow

2.0 开始安装

首先创建一个虚拟环境

我选择2.3tensorflow+3.8python

conda create -n tf2.3 python=3.8

然后激活环境进行下一步安装（切记！）

conda install cudatoolkit=10.1

conda install cudnn=7.6

pip install tensorflow==2.3

验证是否安装成功

python

import tensorflow as tf

tf.__version__

验证GPU是否可用

import tensorflow as tf
tf.config.list_physical_devices('GPU')

[PhysicalDevice(name=’/physical_device:GPU:0’, device_type=‘GPU’)]

参考文章

conda安装tensorflow（python）.
anaconda安装TensorFlow GPU版本（2021.11 3060亲测）.