配置:
Tensorflow-gpu 2.0 alpha
CUDA 10.0 CUDNN7.5
WINDOWS 10
Python 3.6
刚开始跑一般的神经网络出现了个这么个问题,直接用下面链接的方法
https://blog.csdn.net/weixin_45783552/article/details/103025198?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1
即在开头加入下面几句代码
physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)
如果你是1.X的版本
有各种其他朋友写的方法,很好用
另外可能原因:
如果你是Jupyter notebook写的
更大的可能性是你同时打开了好几个文件
由于自己后来跑卷积神经网络,它在计算卷积的时候又报了这么个错,但之前的解决方法行不通
一种方法:
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘/gpu:0’
直接不用GPU跑,省事
自己的解决方法
升级了一下CUDNN版本,升成了7.6.1,再加上了前面的那段代码,完事