要在Linux上成功安裝PyTorch,請按照以下步驟操作:
python --version
pip --version
如果這些命令返回版本號,那么Python和pip已經安裝。如果沒有,請先安裝Python和pip。
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 -m venv pytorch_env
source pytorch_env/bin/activate
對于CPU版本:
pip install torch torchvision torchaudio
對于支持CUDA 11.3的系統:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
請根據您的實際情況替換上述命令中的CUDA版本號。
python -c "import torch; print(torch.__version__)"
如果成功輸出了PyTorch的版本號,那么恭喜您,已經成功安裝了PyTorch!