PyTorch-GPU加速实例

硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1将数据和网络都推到GPU，接上.cuda()你的电脑里面有合适的GPU显卡,并且需要支持CUDA模块你必须安装GPU版的Torch，利用pytorch使用GPU进行加速方法主要就是将数据的形式变成GPU能读的形式，然后将CNN也变成GPU能读的形式，具体办法就是在后面加上.cuda()。