Как установить PyTorch v1.0.0 + в Google Colab?

Question

PyTorch v1.0.0 стабильной была выпущена 8 декабря 2018 года после того, как была объявлена ​​ 7 месяцами ранее .

Я хочу получить версию, оптимизированную для оборудования, которое мойЯдро IPython работает.

Как мне получить эту версию в Google Colab?

Answer 1

попробуйте следующий фрагмент кода (он работает одинаково для среды выполнения с или без gpu)

!pip install -q torch==1.0.0 torchvision

, чтобы проверить версию

import torch
print(torch.__version__)

здесь у вас есть версия 1.0.0

ОБНОВЛЕНИЕ

!pip install torch

Теперь работает нормально, так как самая стабильная версия - 1.0.0

Answer 2

Для версии 1.1.0 это работает

!pip install -q torch==1.1.0 torchvision

Answer 3

В версии 1.0.0 PyTorch изменил формат URL-адреса для загрузки с:

https://download.pytorch.org/whl/cu92/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

на

https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

(Изменение относится к части версии CUDA, где cu92изменяется на cu90.)

Чтобы программно сгенерировать этот URL, я использовал следующий код:

from os.path import exists
from wheel.pep425tags import get_abbr_impl, get_impl_ver, get_abi_tag

platform = '{}{}-{}'.format(get_abbr_impl(), get_impl_ver(), get_abi_tag())
cuda_output = !ldconfig -p|grep cudart.so|sed -e 's/.*\.\([0-9]*\)\.\([0-9]*\)$/cu\10/'    
accelerator = cuda_output[0] if exists('/dev/nvidia0') else 'cpu'

torch_url=f"http://download.pytorch.org/whl/{accelerator}/torch-{version}-{platform}-linux_x86_64.whl"
version='1.0.0'

!pip install -U {torch_url} torchvision

Затем вы можете изменить переменную version по желанию по мере выпуска новых версий.