第一步！指定gpu
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
第二步！
对于每一个要踹到gpu去的Tensor或者model x
使用x = x.cuda()就ok了
嘤嘤嘤
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
第一步！指定gpu
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
第二步！
对于每一个要踹到gpu去的Tensor或者model x
使用x = x.cuda()就ok了
嘤嘤嘤
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。