一、Tensorflow2 GPU 安装方法
- 1. 首先安装Anaconda3环境
- 2. 在Anaconda Prompt 中安装tensorflow2
- 3. 验证GPU是否可以使用
- 4. 错误解决
1. 首先安装Anaconda3环境
https://www.anaconda.com/
2. 在Anaconda Prompt 中安装tensorflow2
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.
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中就可以了