Pytorch是Facebook人工智能研究院基于Torch开发的一个开源的Python机器学习库，它提供了两项强大的功能：

1、强大的GPU加速的张量计算（NumPy)。

2、自动求导系统的深度神经网络。

cuda 11.8

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

cuda 12.1

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121

cpu 

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cpu

Previous PyTorch Versions | PyTorch