Установите Pytorch из источника с помощью pip - PullRequest
2 голосов
/ 30 октября 2019

Я пытаюсь установить pytorch на удаленном сервере. Он имеет CentOS 6.5 и согласно этой ссылке он прекратил поддержку CentOS 6. Итак, я пытаюсь установить его через source .
Рекомендованный метод установки - anaconda, но дело в том, что у меня возникает множество проблем при установке anaconda, так как он часто портит путь к удаленному серверу, поэтому я решил использовать pip.
Но у меня есть проблемы, связанные спреобразование некоторых команд conda в pip, как показано ниже -

conda install -c pytorch magma-cuda90
Указанная выше команда упоминается перед этапом клонирования pytorch, и она выдала мне ошибку, что
Could not open requirements file: [Errno 2] No such file or directory: 'pytorch'
Другая проблема, с которой я сталкиваюсь, ниже -
export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"} Какая альтернатива для CMAKE_PREFIX_PATH в пипсах?

1 Ответ

3 голосов
/ 30 октября 2019

В соответствии с вашей версией Python, вы можете попробовать установить из файлов wheel .

pip install https://download.pytorch.org/whl/cu101/torch-1.3.0-cp36-cp36m-manylinux1_x86_64.whl   --user # For torch 
pip install https://download.pytorch.org/whl/cu101/torchvision-0.4.1-cp36-cp36m-linux_x86_64.whl  --user # For torchvision

В случае сбоя вы можете проверить версию glibc:

ldd --version

, потому что PyTorch поддерживается в дистрибутивах Linux, использующих glibc> = v2.17.

На ваш вопрос:

Какая альтернатива для `CMAKE_PREFIX_PATH в pip?

CMAKE_PREFIX_PATH действует как директива сборки, указывающая, где найти модули, необходимые для сборки. В вашем случае (установка без полномочий root с флагом --user) это, вероятно,:

~/.local/lib/python3.6/site-packages

В качестве примечания, ваша компиляция, скорее всего, не удастся, если вы все еще этого не сделаете. иметь минимально требуемую версию glibc.

...