第一步！指定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刪除。