python - 使用 pip 从源安装 pytorch

Question

我正在尝试在远程服务器上安装 pytorch。它有 CentOS 6.5 并且根据 this link它已停止支持 CentOS 6。所以，我正在尝试通过 source 安装它.
推荐的安装方法是通过 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))/../"}pip 中 CMAKE_PREFIX_PATH 的替代方案应该是什么？

Answer 1

根据你的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 在使用 glibc >= v2.17 的 Linux 发行版上受支持。

针对您的问题:

What should be the alternative for `CMAKE_PREFIX_PATH in pip ?

CMAKE_PREFIX_PATH充当构建指令以指示在哪里可以找到构建所需的模块。在您的情况下(使用 --user 标志作为非 root 用户安装)可能是:

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

您可以使用以下命令验证确切位置:

python -c "import site; print(site.getsitepackages()[0])"

作为旁注，如果您仍然没有最低要求的 glibc 版本，您的编译很可能会失败。