码迷,mamicode.com
首页 > 其他好文 > 详细

TensorFlow报错:failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE

时间:2017-09-06 14:37:17      阅读:958      评论:0      收藏:0      [点我收藏+]

标签: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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!