产生这个错误的原因网上千奇百怪,后来我发现我是因为没有给pycharm分配gpu空间
在代码前面加上这个分配gpu内存的命名一般就可以解决
os.environ[“CUDA_VISIBLE_DEVICES”] = “PCI_BUS_ID”
os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0’
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9
session = tf.Session(config=config)
第二种方法:
from tensorflow import ConfigProto
from tensorflow import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
一般第一种就可以解决,如果没有就用第二种
第三种究极方法,一针见效:
在pycharm里面配置环境,添加环境名LD_LIBRARY_PATH,赋值为你cuda的两个lib64的地址名