os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0’
config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.99
with tf.compat.v1.Session(config=config) as sess:
**该方法需考虑tensorflow版本问题/、
最终解决办法:
将原有cuDNN 7.4.2版本更换为7.6.5
删掉CUDA bin,include,lib文件夹中原有cuDNN 7.4.2版本拷贝的文件,再将cuDNN 7.6.5版本文件拷贝到相应文件夹中。此时可直接运行Keras和Tensorflow框架中的代码,没有报错。
是将7.6.5相应版本复制到文件夹里,不是纯放置