TensorFlow安装CPU版本和GPU版本

席惜兮兮

已于 2024-06-30 02:05:23 修改

阅读量6k

点赞数 40

文章标签： tensorflow 人工智能

于 2024-06-30 01:12:50 首次发布

本文链接：https://blog.csdn.net/Huanghuali_/article/details/140070076

版权

文章目录

前言
一、TensorFlow安装CPU版本
二、TensorFlow安装GPU版本

前言

下载的Anaconda是Anaconda3-2024.02-1-Windows-x86_64版本

一、TensorFlow安装CPU版本

本例子，下载的Python版本为3.11.7和tensorflow版本为2.16.1

1.新建虚拟环境

打开Anaconda Prompt，输入

conda create -n myenvname python=3.11.7

“myenvname”为自己的虚拟环境名字
在这里插入图片描述

2.激活虚拟环境

继续输入

activate myenvname

“myenvname”为自己的虚拟环境名字
在这里插入图片描述

3.下载tensorflow

直接安装tensorflow会遇到以下报错，这是提示有一些依赖没有安装
在这里插入图片描述
所以我先安装了依赖再下载tensorflow

pip install joblib==1.2.0 scipy==1.11.4 tabulate==0.9.0 tqdm==4.65.0 tensorflow==2.16.1 -i https://mirrors.aliyun.com/pypi/simple

在这里插入图片描述

4.验证是否下载成功

输入ipython，进入交互环境（要是报错，那可能是没有ipython，可以pip list查看一下，没有的话需要下载一个）
导入tensorflow

import tensorflow as tf

在这里插入图片描述
成功

二、TensorFlow安装GPU版本

本例子，下载的CUDA版本是11.5.2，cuDNN的版本是8.3.2，Python环境是3.9，tensorflow-gpu的版本是2.7.0。注：CUDA、cuDNN、python的环境要对应，不然会安装失败（很重要！！！）
CUDA和cuDNN具体安装教程看这篇GPU配置环境，这个教程很详细的介绍了GPU的配置。