pytorch 安装部署（cpu+gpu） linux+windows

作者：System 时间：2024年08月13日分类：所有,linux 字数：1061

在Linux和Windows上安装PyTorch通常可以通过Python的包安装工具pip完成。以下是基于CUDA版本的PyTorch安装命令。

对于Linux系统：

如果你没有NVIDIA GPU或者不需要CUDA支持，可以使用：
```
pip install torch torchvision torchaudio
```
如果你需要特定版本的CUDA支持，可以在PyTorch的官方网站上找到相应的安装命令：PyTorch Get Started。例如，对于CUDA 11.1，命令如下：
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu113/torch_stable.html
```

对于Windows系统：

如果你没有NVIDIA GPU或者不需要CUDA支持，可以使用：
```
pip install torch torchvision torchaudio
```
如果你需要CUDA支持，请确保你安装了合适版本的Visual Studio Redistributable和CUDA Toolkit，并且系统的PATH环境变量中包含了CUDA的bin目录。然后，你可以在PyTorch Get Started页面找到相应的安装命令。例如，对于CUDA 11.1，命令如下：
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu113/torch_stable.html
```

注意：确保你的pip是最新版本，以便能够找到与你的系统和CUDA版本兼容的wheel文件。如果需要更新pip，可以使用以下命令：




pip install --upgrade pip