TensorFlow 报错 failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE
解决方法转抄如下:
原因是在创建session时没有使用我想让它用的gpu
所以需要设置
CUDA_VISIBLE_DEVICES
方法1 在python程序中
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “1”
方法2
运行时
CUDA_VISIBLE_DEVICES=1 python **.py
方法3
直接在环境变量中 export