TensorFlow 报错 failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE

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

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据