Как установить PyTorch v1.0.0+ в Google Colab?
PyTorch v1.0.0 стабильный был выпущен 8 декабря 2018 года после того, как было объявлено 7 месяцами ранее.
Я хочу получить версию, оптимизированную для оборудования, на котором работает мое ядро IPython.
Как мне получить эту версию в Google Colab?
2 ответа
Попробуйте следующий фрагмент кода (он работает одинаково для среды выполнения с или без gpu)
!pip install -q torch==1.0.0 torchvision
проверить версию
import torch
print(torch.__version__)
здесь у вас есть версия 1.0.0
ОБНОВИТЬ
!pip install torch
Теперь работает нормально, так как самая стабильная версия 1.0.0
В версии 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 переменная по желанию, как новые версии выпущены.
Теперь ты можешь просто
import torch
Не требует дополнительной установки.
Для версии 1.1.0 это работает
!pip install -q torch==1.1.0 torchvision
Это сработало для меня, вы можете попробовать
!pip install torch
!pip install torchvision
!pip install mxnet-cu101
Вот код для установки pytorch. Вы можете изменить его на любую желаемую версию.
!pip3 install http://download.pytorch.org/whl/cu92/torch-1.0.0-cp36-cp36m-linux_x86_64.whl