参考:

最全Tensorflow安装教程

查看python,cuDNN,CUDA,Trnsorflow的对应关系,网址如下:https://tensorflow.google.cn/install/source?hl=zh-cn#gpu

eg:

# 1.创建虚拟环境
conda create -n tensorflow python=3.8 


# 2.查看下载源中的包的各个版本,根据包含的版本号,指定下载版本
## -c conda-forge,指定搜索渠道conda-forge,查看包含的软件包的版本
conda search tensorflow-gpu -c conda-forge 
conda search cudnn -c conda-forge 
conda search cudatoolkit -c conda-forge

# 3.安装指定版本的tensorflow-gpu
## conda会自动安装匹配tensorflow版本的cudatoolkit和cudnn,不需要手动安装
conda install tensorflow-gpu==2.6.0 -c conda-forge

# 4.检测是否安装成功
import tensorflow as tf # 导入成功,说明tensorflow下载正确
print(tf.test.is_gpu_available()) # 返回True,则GPU可用

更多推荐