Pytorch使用多GPU

在caffe中訓練的時候如果使用多GPU則直接在運行程序的時候指定GPU的index即可，但是在Pytorch中則需要在聲明模型之后，對聲明的模型進行初始化，如： cnn = DataPara ...