pytorch 安装（转载）

使用CUDA安装9

conda install pytorch=0.4.1 cuda90 -c pytorch

要么

conda install pytorch=0.4.1 cuda92 -c pytorch

使用CUDA安装8

conda install pytorch=0.4.1 cuda80 -c pytorch

使用CUDA 7.5进行安装

conda install pytorch=0.4.1 cuda75 -c pytorch

没有CUDA安装

conda install pytorch=0.4.1 -c pytorch

从source

可以签出旧版本的PyTorch 并构建它。您可以在PyTorch git存储库中列出标签git tag并检出特定的标签（用所需版本替换“0.1.9”）

git checkout v0.1.9

按照PyTorch checkout的README.md中的源代码说明进行安装。

通过pip

whl通过此命令下载具有所需版本的文件（您可以使用您选择的版本替换1.0.1）：

pip install torch==1.0.1 -f https://download.pytorch.org/whl/cpu/stable # CPU-only buildpip install torch==1.0.1 -f https://download.pytorch.org/whl/cu80/stable # CUDA 8.0 buildpip install torch==1.0.1 -f https://download.pytorch.org/whl/cu90/stable # CUDA 9.0 buildpip install torch==1.0.1 -f https://download.pytorch.org/whl/cu92/stable # CUDA 9.2 buildpip install torch==1.0.1 -f https://download.pytorch.org/whl/cu100/stable # CUDA 10.0 build

注意：大多数pytorch版本仅适用于特定的CUDA版本。例如，pytorch = 1.0.1不适用于CUDA 9.2