tensorflow.python.framework.errors_impl.InternalError: Failed to create session.解决办法汇总
最近,写了一个简单的tensorflow程序来测试新的电脑。由于新电脑是两个不同的nvidia显卡,所以,每次都需要设置一下到底使用哪个gpu,要不然老是弹出无法创建会话的错误,无法创建会话的错误似乎有好几种。
解决办法:
在程序中添加以下代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0' #使用编号为0的gpu
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
config.gpu_options.allow_growth = True
这个方法可以解决上面图片中的错误
其他无法创建会话的解决办法:
在运行程序时将python test.py
改为CUDA_VISIBLE_DEVICES=0 python test.py