码迷,mamicode.com
首页 > 编程语言 > 详细

深度学习 多卡 python设置

时间:2017-05-22 10:25:44      阅读:570      评论:0      收藏:0      [点我收藏+]

标签:imp   visible   pytho   devices   port   使用   深度   深度学习   cuda   

首先 import os

然后通过 os.environ[‘CUDA_VISIBLE_DEVICES‘]=来设置用哪张卡

比如使用编号为0的卡:import os
                                     os.environ[‘CUDA_VISIBLE_DEVICES‘]=‘0‘

深度学习 多卡 python设置

标签:imp   visible   pytho   devices   port   使用   深度   深度学习   cuda   

原文地址:http://www.cnblogs.com/yxxx/p/6888215.html

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