Запуск и сборка Pytorch в Google Colab

Question

Я пытаюсь запустить пакет python, требующий pytorch-gpu. Я изменил тип среды выполнения моего ноутбука Colab на GPU. Когда я запускаю команду, я сталкиваюсь со следующей ошибкой. Не уверены, смогу ли я самостоятельно построить pytorch на colab?

Traceback (most recent call last):
  File "inference_unet.py", line 9, in <module>
    import torchvision.transforms as transforms
  File "/usr/local/lib/python3.6/dist-packages/torchvision/__init__.py", line 10, in <module>
    from .extension import _HAS_OPS
  File "/usr/local/lib/python3.6/dist-packages/torchvision/extension.py", line 58, in <module>
    _check_cuda_version()
  File "/usr/local/lib/python3.6/dist-packages/torchvision/extension.py", line 54, in _check_cuda_version
    .format(t_major, t_minor, tv_major, tv_minor))
RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=10.2 and torchvision has CUDA Version=10.1. Please reinstall the torchvision that matches your PyTorch install.

Answer 1

Теперь вы можете напрямую использовать pytorch-gpu в google colab, без установки. Просто измените среду выполнения на gpu, импортируйте torch и torchvision, и все готово. Я приложил снимок экрана, сделав то же самое.

Надеюсь, ответ окажется полезным.

Но если вы хотите установить другую версию pytorch или любого другого пакета, затем вы можете установить с помощью pip, просто добавьте! перед командой pip и запустите ячейку.

например,