Tensorflow2 GPU 安装方法

坐在墙上的猫2023-10-04 0:35

一、Tensorflow2 GPU 安装方法

- [1. 首先安装Anaconda3环境](#1. 首先安装Anaconda3环境)
- [2. 在Anaconda Prompt 中安装tensorflow2](#2. 在Anaconda Prompt 中安装tensorflow2)
- [3. 验证GPU是否可以使用](#3. 验证GPU是否可以使用)
- [4. 错误解决](#4. 错误解决)

1. 首先安装Anaconda3环境

https://www.anaconda.com/

2. 在Anaconda Prompt 中安装tensorflow2

bash 复制代码

conda update conda
conda create -n tensorflow python=3.9
conda activate tensorflow
pip install tensorflow-gpu==2.7.0 -i  https://pypi.mirrors.ustc.edu.cn/simple
pip install protobuf==3.19.6 -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn

3. 验证GPU是否可以使用

参考博客：https://blog.csdn.net/weixin_43412762/article/details/129824339

4. 错误解决

如果在训练时出现以下错误，需要重新安装其他版本的cuda,我是重新安装了cuda12的cuda.

bash 复制代码

 tensorflow/stream_executor/gpu/redzone_allocator.cc:314] UNKNOWN: Failed to create a NewWriteableFile:

如果出现 I tensorflow/stream_executor/cuda/cuda_dnn.cc:366] Loaded cuDNN version 8500，然后程序终止了，提示缺少文件

按照图片进行配置，可以看到实际的报错是什么，会在输出中看到缺少安装zlibwapi.lib。

下面是链接

下载之后放在C:\Windows\System32中就可以了