使用 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

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

针对您的问题：

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 版本，您的编译很可能会失败。