要在Linux上成功安装PyTorch,请按照以下步骤操作:
- 首先,确保您的系统已安装了Python和pip。您可以通过在终端中运行以下命令来检查它们是否已安装:
python --version
pip --version
如果这些命令返回版本号,那么Python和pip已经安装。如果没有,请先安装Python和pip。
- 接下来,打开终端并运行以下命令以添加PyTorch的官方GPG密钥:
curl -sSL https://download.pytorch.org/whl/cu113/torch_stable.html | grep -A 2 "apt-key" | awk '{print $4}' | sudo apt-key add -
请注意,上述命令中的cu113
表示针对CUDA 11.3的PyTorch版本。如果您使用的是其他版本的CUDA,请将cu113
替换为相应的版本号。您可以在PyTorch官方网站上找到适用于您的系统的命令。
- 创建一个新的Python虚拟环境(可选,但推荐):
python -m venv pytorch_env
source pytorch_env/bin/activate
- 使用pip安装PyTorch。根据您的系统配置和CUDA版本,运行以下命令之一:
对于CPU版本:
pip install torch torchvision torchaudio
对于支持CUDA 11.3的系统:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
请根据您的实际情况替换上述命令中的CUDA版本号。
- 安装完成后,验证PyTorch是否已成功安装:
python -c "import torch; print(torch.__version__)"
如果成功输出了PyTorch的版本号,那么恭喜您,已经成功安装了PyTorch!