PyTorch 安装命令合集

Torch 1.0.0 版本将下文中torch-0.4.1更换为torch-1.0.0

Linux

使用 conda
以下命令适用于 Python 2.7 - 3.7。

CPU 版本:
conda install pytorch-cpu torchvision-cpu -c pytorch

CUDA 8:
conda install pytorch torchvision cuda80 -c pytorch

CUDA 9.0:
conda install pytorch torchvision -c pytorch

CUDA 9.2:
conda install pytorch torchvision cuda92 -c pytorch

使用 pip
Python 2.7

CPU 版本:
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
注:if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27m-linux_x86_64.whl

CUDA 8:
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
注:if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp27-cp27m-linux_x86_64.whl

CUDA 9.0:
pip install torch torchvision

CUDA 9.2:
pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
注:if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp27-cp27m-linux_x86_64.whl

Python 3.5
CPU 版本:
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

CUDA 8:
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

CUDA 9.0:
pip3 install torch torchvision

CUDA 9.2:
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

Python 3.6
CPU 版本:
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

CUDA 8:
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

CUDA 9.0:
pip3 install torch torchvision

CUDA 9.2:
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

Python 3.7
CPU 版本:
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

CUDA 8:
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

CUDA 9.0:
pip3 install torch torchvision

CUDA 9.2:
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

MacOS

使用 conda
以下命令适用于 Python 2.7 - 3.7。

CPU 版本:
conda install pytorch torchvision -c pytorch
CUDA 8,CUDA 9.0,或 CUDA 9.2:MacOS 不支持 CUDA,如果需要,则需要从源码编译安装。

使用 pip
Python 2.7
CPU 版本:
pip install torch torchvision
CUDA 8,CUDA 9.0,或 CUDA 9.2:MacOS 不支持 CUDA,如果需要,则需要从源码编译安装。

Python 3.5,3.6,3.7
CPU 版本:
pip3 install torch torchvision
CUDA 8,CUDA 9.0,或 CUDA 9.2:MacOS 不支持 CUDA,如果需要,则需要从源码编译安装。

Windows

使用 conda
Python 2.7
PyTorch 不支持在 Windows 上的 Python 2.7 版本,请使用 Python 3 版本。

Python 3.5,3.6,3.7
CPU 版本:
conda install pytorch-cpu -c pytorch
pip3 install torchvision

CUDA 8:
conda install pytorch cuda80 -c pytorch
pip3 install torchvision

CUDA 9.0:
conda install pytorch -c pytorch
pip3 install torchvision

CUDA 9.2:
conda install pytorch cuda92 -c pytorch
pip3 install torchvision

使用 pip
Python 2.7
PyTorch 不支持在 Windows 上的 Python 2.7 版本,请使用 Python 3 版本。

Python 3.5
CPU 版本:
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

CUDA 8:
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

CUDA 9.0:
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

CUDA 9.2:
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

Python 3.6
CPU 版本:
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

CUDA 8:
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

CUDA 9.0:
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

CUDA 9.2:
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

Python 3.7
CPU 版本:
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

CUDA 8:
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

CUDA 9.0:
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

CUDA 9.2:
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

### 回答1: 可以使用 Anaconda 的命令行工具 conda 来安装 PyTorch。在命令行中运行以下命令即可安装 PyTorch: ``` conda install pytorch torchvision -c pytorch ``` 如果你需要指定安装特定版本的 PyTorch,可以在上面的命令中加入版本号,例如: ``` conda install pytorch==1.7 torchvision==0.8 -c pytorch ``` 安装完成后,可以通过在 Python 中运行以下代码来确认安装是否成功: ``` import torch print(torch.__version__) ``` 如果你运行上述代码之后,没有报错且能够看到 PyTorch 的版本号,说明安装成功了。 ### 回答2: Anaconda作为常用的Python科学计算环境之一,其内置了强大的Python包管理器Conda,可以方便地安装和管理许多常用的科学计算库,包括PyTorch安装步骤如下: 1. 首先需要安装anaconda环境,可以从官网下载对应系统的安装包,然后进行安装。 2. 打开终端(Terminal),输入以下命令来创建一个新的环境并激活之: ``` conda create --name torch_env python=3.6 conda activate torch_env ``` 其中`torch_env`为新环境的名称,可以更改为自己喜欢的名称。 3. 接下来,输入以下命令安装PyTorch: ``` conda install pytorch torchvision -c pytorch ``` 这里`-c pytorch`是指从PyTorch的官方conda channel中下载安装包。 4. 等待安装完成后,可以通过输入以下命令来验证PyTorch是否成功安装: ``` python import torch print(torch.__version__) ``` 如果输出了PyTorch的版本号,则说明安装成功。 此外,还可以安装一些常用的第三方库,比如numpy、pandas、scipy等,具体如下: ``` conda install numpy pandas scipy matplotlib scikit-learn ``` 至此,完整的安装过程就完成了。在使用PyTorch时,记得先激活对应的环境,然后再引入相应的包即可。 ### 回答3: Anaconda是一个流行的Python发行版,它包含了许多常用的数据科学工具和库。如果你想要在Anaconda环境中安装PyTorch,则需要按照以下步骤进行操作。 1.打开Anaconda Navigator,找到左侧菜单栏中的“Environments”选项卡,点击“Create”创建一个新的Python环境。 2.给这个新的环境起一个名字,并且选择需要安装的Python版本,这里建议选择3.X的版本。 3.在新的环境中打开“Terminal”,在命令行窗口中输入以下命令安装PyTorch: ```python conda install pytorch torchvision torchaudio cpuonly -c pytorch ``` 这个命令安装PyTorch和相关的库。如果你的电脑使用的是Nvidia GPU,则可以将“cpuonly”替换为“cudatoolkit=10.2”来安装支持GPU的版本。 4.等待一段时间,直到安装完成。 5.测试安装是否成功,可以在命令行窗口中输入以下Python代码: ```python import torch print(torch.__version__) ``` 如果输出的版本号和你安装的版本号一致,则表示安装成功。 总的来说,使用Anaconda安装PyTorch非常简单。通过这种方式,你可以轻松地管理你的环境以及安装所需的库和工具。如果你遇到了任何问题,可以在StackOverflow等社区向其他开发者寻求帮助。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值