pytorch 安装(转载)

2019-06-20  本文已影响0人  小小兰哈哈

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

上一篇 下一篇

猜你喜欢

热点阅读