安装PyTorch

torch 是 PyTorch 的核心库，PyTorch 是一个流行的开源深度学习框架，由 Facebook 的人工智能研究团队开发。它主要用于以下几个方面：
  • 张量计算（类似于NumPy）：支持GPU加速。
  • 深度神经网络的构建和训练：提供了丰富的API来定义和训练深度学习模型。
  • 自动微分：支持动态计算图的自动微分，用于梯度计算。
PyTorch的安装可以通过pip或conda进行，具体的安装命令取决于你的操作系统、Python版本以及是否需要GPU支持。

使用conda安装

如果你使用的是Anaconda环境，可以使用以下命令安装CPU版本：
  • conda install pytorch torchvision torchaudio cpuonly -c pytorch
如果你需要GPU支持（CUDA），可以使用以下命令：
  • conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

使用pip安装

如果你使用的是CPU版本，可以使用以下命令安装：
  • pip install torch torchvision torchaudio
如果你需要GPU支持（CUDA），需要到PyTorch的官网生成适合你系统的安装命令。以下是一个示例，假设你使用的是CUDA 11.7：