Anaconda3 安装 Tensorflow-gpu

一.准备需要

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)

上一篇
Window10搭建GPU环境(CUDA、cuDNN)
下一篇
1.1 机器学习是什么(TensorFlow )

猜你喜欢

转载自blog.csdn.net/qq_33253054/article/details/130414379