PyTorch 是一个流行的开源机器学习库，广泛用于深度学习研究和开发。以下是在 Linux 系统上安装 PyTorch 的详细步骤。

系统要求

  • Ubuntu 16.04/18.04/20.04
  • CentOS 7
  • Debian 9

安装步骤

  1. 更新系统包
sudo apt-get update
sudo apt-get upgrade
  1. 安装依赖
sudo apt-get install -y python3-pip python3-dev build-essential libopenblas-dev
  1. 安装 PyTorch

首先，你需要确定你的 Python 版本和硬件支持。以下是一些安装命令的例子：

  • CPU 版本：
pip3 install torch torchvision torchaudio
  • GPU 版本：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
  1. 验证安装

安装完成后，可以通过以下命令验证 PyTorch 是否安装成功：

python3 -c "import torch; print(torch.__version__)"

注意事项

  • 确保你的 Python 环境是干净且唯一的，避免不同项目之间的依赖冲突。
  • 如果你需要使用 GPU 加速，请确保你的显卡支持 CUDA。

扩展阅读

更多关于 PyTorch 的安装和使用信息，请访问我们的 PyTorch 安装指南。

图片示例

PyTorch 在 Linux 系统上的安装界面

Linux PyTorch Installation Screen