python中使用指定GPU

时间:2018-10-30 14:24:33   收藏:0   阅读:585

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "2"

 

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

原文:https://www.cnblogs.com/sbj123456789/p/9876474.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!