主要介绍了PyTorch-GPU加速实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

硬件：NVIDIA-GTX1080 软件：Windows7、python3.6.5、pytorch-gpu-0.4.1 一、基础知识 将数据和网络都推到GPU，接上.cuda() 二、代码展示 import torch import torch.nn as nn import torch.utils.data as Data import torchvision # torch.manual_seed(1) EPOCH = 1 BATCH_SIZE = 50 LR = 0.001 DOWNLO

代码已经调通，跑出来的效果如下： # coding=gbk import torch import matplotlib.pyplot as plt from torch.autograd import Variable import torch.nn.functional as F ''' Pytorch是一个拥有强力GPU加速的张量和动态构建网络的库，其主要构建是张量，所以可以把PyTorch当做Numpy 来用,Pytorch的很多操作好比Numpy都是类似的，但是其能够在GPU上