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

发表评论

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

This site uses Akismet to reduce spam. Learn how your comment data is processed.