一.准备需要
GPU版本,需要提前下载 cuda 和 cuDNN。(可以查看此教程 Anaconda3 安装 Tensorflow-gpu )
GPU 一般是 nvidia 的
安装前 一定 要查看自己电脑的环境配置,然后查询Tensorflow-gpu、Python、 cuda 、 cuDNN 版本关系,要 一 一对应!
Tensorflow-gpu 与 Python、 cuda、cuDNN 版本关系查询
二.安装 Tensorflow-gpu
创建环境
或者
conda create -n tensorflow python=3.10.0
查看环境,可以看到环境已创建**
conda env list
进入环境
activate tensorflow
进入具体路径
# 这里换成自己的具体路径 可以在 conda env list里面找到
>cd C:\Users\Lenovo\anaconda3\envs\tensorflow
可以查看 Tensorflow-gpu有哪些版本
pip install tensorflow-gpu ==
安装 Tensorflow-gpu 的2.10.0版本
#安装对应的版本
pip install tensorflow-gpu ==2.10.0
三.查看 Tensorflow-gpu是否运行
import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())
参考资料
TensorFlow安装步骤
Tensorflow-gpu保姆级安装教程(Win11, Anaconda3,Python3.9)