在做手写字符识别时遇到的问题
原本环境:WIN11+CUDA11.0+CUDNN8.0+anaconda3+tensorflow-gpu2.4.0
问题:在训练时一直报错Call to CreateProcess failed. Error code: 2,但是对结果好像并没有影响。
解决方法:将环境更改为CUDA10.1+CUDNN7.6.5+anaconda3+tensorflow-gpu2.3.0后不再报错。(CUDA向下是兼容的)
补充:tfgpu2.4使用pip直接安装时很容易出现缺少文件的情况,且有很多奇怪的bug,故不推荐使用。
tfgpu的安装教程:
1.创建新环境
conda create -n tf23 python=3.82.激活环境
conda activate tf233.安装cudatoolkit与cudnn
conda install cudatoolkit=10.1
conda install cudnn=7.6.54.安装tfgpu2.3
pip install tensorflow==2.3.0 -i https://pypi.douban.com/simple/5.用pycharm加载该虚拟环境
6.新建py文件进行测试
import tensorflow as tf
gpus = tf.config.list_physical_devices("GPU")
print(gpus)
if gpus:
gpu0 = gpus[0] # 如果有多个GPU,仅使用第0个GPU
tf.config.experimental.set_memory_growth(gpu0, True) # 设置GPU显存用量按需使用
tf.config.set_visible_devices([gpu0], "GPU")

版权声明:本文为qq_43732541原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。