Как включить поддержку графического процессора pytorch в Google Colab?

Question

Как я могу включить pytorch для работы на GPU?

Я успешно установил pytorch в google colab notebook: Tensorflow сообщает о наличии графического процессора:

Но функция torch.device как-то не работает:

Как я могуисправить это?

Answer 1

Вы можете использовать этот учебник: 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:

torch.cuda.get_device_name(0)

Мой результат в Google Colab - Tesla K80.

Answer 2

Я столкнулся с той же проблемой.

Попробуйте установить 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 Accelerator: cu80 Torch 0.4.0 CUDA 8.0.61
Устройство: cuda: 0

Некоторые фрагменты, плавающие вокруг, используют torch-0.3.0.post4-{platform}-linux_x86_64.whl, что приведет к той же ошибке, поскольку device являетсяФакел 4 особенность.Если вы уже установили неправильную версию, вам может потребоваться !pip uninstall torch.

. Обязательно включите графический процессор в разделе Правка> Настройки ноутбука> Аппаратный ускоритель .