Как использовать GPU в pytorch? - PullRequest
Новая
       30

Как использовать GPU в pytorch?

0 голосов
/ 06 февраля 2020

Я попытался выполнить следующие шаги: https://pytorch.org/get-started/locally/

Сначала я создал среду conda следующим образом: 

conda create -n facenet37_2 python=3.7

Затем на указанном выше сайте я выбрал: 

PyTorch Build: Stable (1.4)
OS: Linux (I am using Ubuntu 18.04)
Package: conda
Language: python
CUDA: 10.1

и он попросил меня выполнить следующую команду: 

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

Но после этого, когда я открыл python и набрал: 

import torch
torch.cuda.is_available()

Я получил False

У меня GeForce GT 630M (computeCapability: 2.1). Но это не обнаруживается. Почему? Это слишком старое и больше не поддерживается? Как я могу решить проблему?

Редактировать: Почему я получил отрицательный голос?

Ответы [ 2 ]

1 голос
/ 07 февраля 2020

GeForce GT 630M имеет вычислительные возможности 2.1 и поэтому поддерживает только до CUDA 8.

  • PyTorch двоичные файлы исключена поддержка вычислительных возможностей <= 5.0 в <a href="https://github.com/pytorch/pytorch/releases/tag/v0.3.1" rel="nofollow noreferrer"> PyTorch 0.3.1 . Мне не ясно, была ли когда-либо включена возможность вычисления 2.1 в двоичные файлы.
  • База кода PyTorch отказалась от поддержки CUDA 8 в PyTorch 1.1.0 .

Из-за второго момента нет никакой возможности изменить кодовую базу PyTorch, чтобы ваш графический процессор работал с последней версией. Возможны следующие варианты:

  • Установить PyTorch без поддержки графического процессора.
  • Попробуйте скомпилировать PyTorch <1.1.0 из исходного кода (<a href="https://github.com/pytorch/pytorch/blob/v1.0.1/README.md#installation" rel="nofollow noreferrer"> инструкции ). Не забудьте проверить тег v1.0.1. Это создаст двоичный файл с поддержкой ваших вычислительных возможностей.
  • Если это приемлемо, вы можете попробовать установить действительно старую версию: PyTorch <0.3.1, используя conda или колесо, и посмотреть, работает ли это. Возможно, у него есть поддержка вычислений 2.1, но я не могу это проверить. См. <a href="https://pytorch.org/get-started/previous-versions/" rel="nofollow noreferrer"> pytorch.org для информации. Хотя, похоже, что ссылка на https://download.pytorch.org/whl/cu80/torch_stable.html не работает.
0 голосов
/ 06 февраля 2020

Ваша архитектура графического процессора - Fermi. Последняя версия CUDA с поддержкой Fermi: 8.0, и вы установили 10.1

Возможное решение может заключаться в установке более старых версий CUDA и PyTorch. 

conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
...