标签:import pytho 解决 retain ror udev environ 原因 dev
tensorflow-gpu
在创建session时没有使用我想让它用的gpu
1. 在python程序中:
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"
2. 运行时:
CUDA_VISIBLE_DEVICES=0 python **.py
3. 直接在环境变量中 export
export CUDA_VISIBLE_DEVICES=‘0‘ python **.py
TensorFlow报错:failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE
标签:import pytho 解决 retain ror udev environ 原因 dev
原文地址:http://www.cnblogs.com/CongYuUestcer/p/7484278.html