解決方法之一：

如果pytorch在進(jìn)行model.cuda(）操作需要花費(fèi)的時(shí)間很長(zhǎng)，長(zhǎng)到你懷疑GPU的速度了，那就是不正常的。

如果你用的pytorch版本是0.3.0，升級(jí)到0.3.1就好了！

.cuda()加載時(shí)間很長(zhǎng)的其他解決方法

方法一：

pip install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

后面這個(gè)可以自己到官網(wǎng)上找自己對(duì)應(yīng)的python和cuda版本的安裝包，

官網(wǎng)為鏈接：

方法二：

conda install pytorch torchvision cuda80 -c soumith

這里可以根據(jù)自己cuda版本確定。查看cuda版本，在python命令行里面輸入：

import torch
torch.version.cuda

即可。

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。