pytorch 安装

PyTorch 可以通过安装软件包来安装。可以在命令行中使用 pip 进行安装，命令如下:

pipinstall torch torchvision

如果您还需要安装 CUDA 版本的 PyTorch，可以在上述命令的基础上，添加以下内容:

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu22/torch_stable.html

请注意，您需要在安装 CUDA 版本的 PyTorch 之前，先安装 CUDA 和 cuDNN。