引言
在項目的過程中,經常需要復現大牛的代碼。而很多代碼是針對gpu而言的,所以還在cpu環境下跑代碼的人們就會比較頭痛。這個時候一般會有兩種方法:第一種就是嘗試將gpu版本的源代碼修改成cpu版本。這種方法有時候很湊效,就像下面只需要簡單的修改幾行命令即可實現從gpu環境到cpu環境的跨越,或者只是將帶有cudnn(gpu加速)的一些命令去掉即可:
import torch
from torch.autograd import Variable
# 將變數或者數據轉移至GPU
gpu_info = Variable(torch.randn(3,3)).cuda()
# 將變數或者數據轉移至CPU
cpu_info = gpu_info.cpu()
但是有很多時候,一些模塊自帶的命令只支持使用cudnn(見下圖):