如何在Google Colab中启用pytorch GPU支持？

Question

如何启用pytorch在GPU上工作？

我已经在google colab笔记本中成功安装了pytorch： Tensorflow报告GPU已到位：

但是torch.device函数以某种方式失败了：

我该如何解决这个问题？

Answer 1

我遇到了同样的问题。

尝试像这样安装Torch：

# http://pytorch.org/
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' #'cu80' if path.exists('/opt/bin/nvidia-smi') else 'cpu'
print('Platform:', platform, 'Accelerator:', accelerator)

!pip install --upgrade --force-reinstall -q http://download.pytorch.org/whl/{accelerator}/torch-0.4.0-{platform}-linux_x86_64.whl torchvision

import torch
print('Torch', torch.__version__, 'CUDA', torch.version.cuda)
print('Device:', torch.device('cuda:0'))

输出应为：

  

平台：cp36-cp36m加速器：cu80火炬0.4.0 CUDA 8.0.61
  设备：cuda：0

有些摘要使用torch-0.3.0.post4-{platform}-linux_x86_64.whl，这会导致相同的错误，因为device是Torch 4的功能。如果安装的版本错误，则可能需要执行!pip uninstall torch。

还要确保在“编辑”>“笔记本设置”>“硬件加速器” 下启用GPU。

Answer 2

除了在菜单“运行时”->更改运行时类型下启用GPU外，还通过以下方式启用了GPU支持：

import torch

if torch.cuda.is_available():
  device = torch.device("cuda")
else:
  device = torch.device("cpu")

Answer 3

您可以通过单击“运行系统”菜单下的“更改运行系统类型”来启用GPU。这些天也提供“ TPU”支持。

您可以使用device来定义torch.device：

import torch

DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

Answer 4

您可以使用本教程：https://medium.com/@nrezaeis/pytorch-in-google-colab-640e5d166f13

例如CUDA 9.2和Python 3.6：

!pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
!pip3 install torchvision

现在使用PyTorch检查GPU设备：

torch.cuda.get_device_name(0)

我在Google Colab中得到的结果是Tesla K80。