TensorFlow 报错 failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE
解决方案:
问题1:回到主目录,重新运行代码。
问题2:
指定gpu时使用如下任意一种方式,而不是os.environ['CUDA_VISIBLE_DEVICES'] = "1"
# 方法1
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"//一块gpu
# 方法2,在创建session时使用这句代码:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)).