TensorFlow學習筆記 速記2 報錯：failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE

版本：

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