使用pip从源代码安装pytorch

时间:2019-10-29 22:34:06

标签: python pip anaconda pytorch remote-server

我正在尝试在远程服务器上安装pytorch。根据{{​​3}},它具有CentOS 6.5 它已停止支持CentOS 6。因此,我正在尝试通过documentation安装它。
推荐的安装方法是anaconda,但事实是,安装anaconda时会遇到很多问题,因为它困扰着远程服务器路径,因此我决定使用pip。
但是我有如下有关在pip中转换一些conda命令的问题-

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 个答案:

答案 0 :(得分:3)

根据您的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

因为使用glibc> = v2.17的Linux发行版支持PyTorch。

您的问题:

  

pip中的“ CMAKE_PREFIX_PATH”应该替代什么?

CMAKE_PREFIX_PATH充当构建指令,以指示在何处查找构建所需的模块。在您的情况下(带有--user标志的非root用户安装)可能是:

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

请注意,如果您仍然没有glibc的最低要求版本,则编译很可能会失败。