Ubuntu支持多个版本的PyTorch,具体版本取决于所使用的Ubuntu版本以及是否安装了CUDA。以下是一些常见版本的安装方法:
Ubuntu 22.04 LTS
-
CPU版本:可以通过以下命令安装支持CPU的PyTorch版本:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-
GPU版本(需要安装CUDA):要安装支持GPU的PyTorch,请运行以下命令:
pip3 install torch torchvision torchaudio
Ubuntu 18.04
-
CPU版本:
pip3 install torch torchvision torchaudio
Ubuntu 20.04
- GPU版本(需要安装CUDA和cuDNN):可以参考中的指南,根据CUDA版本和Python版本选择合适的PyTorch版本进行离线安装。
请注意,PyTorch的具体版本和安装命令可能会随着Ubuntu版本的更新和PyTorch的发布而发生变化。建议访问PyTorch官网获取最新的安装指南和版本信息。