关键词

pytorch调用gpu

第一步!指定gpu

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'

第二步!

对于每一个要踹到gpu去的Tensor或者model x

使用x = x.cuda()就ok了

嘤嘤嘤

本文链接:http://task.lmcjl.com/news/12604.html

展开阅读全文