小编Moo*_*ech的帖子

我是 PyTorch 的新手，一直在做一些关于 CIFAR10 的教程，特别是使用 Google Colab，因为我个人还没有 GPU 来进行实验。

我已经成功训练了我的神经网络，但我不确定我的代码是否使用 Colab 的 GPU，因为使用 Colab 进行的训练时间并不比我的 2014 MacBook Pro（不带 GPU）快很多。

我检查了一下，我的笔记本确实运行的是 Tesla K80，但不知何故训练速度很慢。所以我想也许我的代码没有配备 GPU 语法，但我无法弄清楚那是哪一部分。

# install PyTorch
from os import path
from wheel.pep425tags import get_abbr_impl, get_impl_ver, get_abi_tag
platform = '{}{}-{}'.format(get_abbr_impl(), get_impl_ver(), get_abi_tag())
accelerator = 'cu80' if path.exists('/opt/bin/nvidia-smi') else 'cpu'
!pip install -q http://download.pytorch.org/whl/{accelerator}/torch-0.4.0-{platform}-linux_x86_64.whl torchvision

import torch
import torch.nn as nn
from torch.optim import Adam
from torchvision import transforms
from torch.autograd import Variable
import torchvision.datasets as datasets
from torch.utils.data import DataLoader, …