溫馨提示×

如何在linux上成功安裝pytorch

小樊
112
2024-12-31 23:33:33
欄目: 智能運維

要在Linux上成功安裝PyTorch,請按照以下步驟操作:

  1. 首先,確保您的系統已安裝了Python和pip。您可以通過在終端中運行以下命令來檢查它們是否已安裝:
python --version
pip --version

如果這些命令返回版本號,那么Python和pip已經安裝。如果沒有,請先安裝Python和pip。

  1. 接下來,打開終端并運行以下命令以添加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官方網站上找到適用于您的系統的命令。

  1. 創建一個新的Python虛擬環境(可選,但推薦):
python -m venv pytorch_env
source pytorch_env/bin/activate
  1. 使用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版本號。

  1. 安裝完成后,驗證PyTorch是否已成功安裝:
python -c "import torch; print(torch.__version__)"

如果成功輸出了PyTorch的版本號,那么恭喜您,已經成功安裝了PyTorch!

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女