PyTorch 安装 - Shell/Bash

PyTorch是一个基于Python的科学计算包，主要针对两类人群：

• 作为NumPy的替代品，可以利用GPU的性能进行计算
• 作为一个高灵活性、速度快的深度学习平台

本文将介绍PyTorch在Shell/Bash环境下的安装方法。

环境

在开始安装之前，请确保您已经安装了以下环境：

• Python（建议使用Python 3.6以上版本）
• pip

安装

在Shell/Bash环境下输入以下命令进行安装：

pip3 install torch torchvision

这将会安装PyTorch及它的图像库torchvision。如果您的系统安装了CUDA，PyTorch也将与CUDA相关联。

要验证您的安装是否成功，请在Python命令行下输入以下内容：

import torch
print(torch.__version__)

如果打印出了您安装的PyTorch版本号，则说明安装成功。

安装指定版本

如果您需要安装PyTorch的指定版本，可以使用以下命令：

pip3 install torch==1.4.0 torchvision==0.5.0

这会安装PyTorch 1.4.0和对应版本的torchvision。请根据需要更改版本号。

安装CPU版PyTorch

如果您的系统没有GPU或者您不需要GPU加速，可以安装CPU版的PyTorch：

pip3 install torch-cpu torchvision

结论

本文介绍了在Shell/Bash环境下安装PyTorch及指定版本的方法。请根据需要安装GPU或CPU版的PyTorch，以便进行深度学习相关的工作。