以下是 PyTorch 在 Linux 系统上的安装步骤,确保您的系统满足以下要求:
- 操作系统:Linux
- Python 版本:Python 3.6, 3.7, 或 3.8
- pip 版本:至少 19.0
安装步骤
更新系统包列表
sudo apt-get update
安装 Python 和 pip
sudo apt-get install python3 python3-pip
安装 PyTorch 打开终端,使用以下命令安装 PyTorch:
pip3 install torch torchvision torchaudio
验证安装 运行以下 Python 代码来验证 PyTorch 是否已正确安装:
import torch print(torch.__version__)
注意事项
- 如果您需要使用 GPU 加速,请确保安装了 CUDA 和 cuDNN。
- 您可以访问 PyTorch 官方文档 获取更多详细信息和安装选项。
PyTorch Logo