办公任务系统
行业管理经验者研发品牌
办公任务分配系统,无需安装,能上网,就能轻松管理!
登录/注册
第一步!指定gpu
import osos.environ["CUDA_VISIBLE_DEVICES"] = '0'
第二步!
对于每一个要踹到gpu去的Tensor或者model x
使用x = x.cuda()就ok了
嘤嘤嘤
本文链接:http://task.lmcjl.com/news/12604.html