溫馨提示×

如何在Ubuntu上運行PyTorch

小樊
84
2025-04-24 01:28:45
欄目: 智能運維

在Ubuntu上運行PyTorch,您需要先安裝Python和pip(如果尚未安裝),然后安裝PyTorch。以下是詳細步驟:

  1. 更新系統包列表:
sudo apt update
  1. 安裝Python3和pip(如果尚未安裝):
sudo apt install python3 python3-pip
  1. 安裝虛擬環境(可選,但推薦):
sudo apt install python3-venv
  1. 創建一個新的虛擬環境(可選):
python3 -m venv pytorch-env
  1. 激活虛擬環境(如果創建了虛擬環境):
source pytorch-env/bin/activate
  1. 根據您的CUDA版本選擇合適的PyTorch安裝命令。您可以在PyTorch官方網站找到適用于Ubuntu的安裝命令:https://pytorch.org/get-started/locally/

例如,如果您想安裝支持CUDA 11.1的PyTorch,可以使用以下命令:

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

如果您不需要GPU支持,可以安裝CPU版本的PyTorch:

pip install torch torchvision torchaudio
  1. 驗證PyTorch是否已成功安裝:
python -c "import torch; print(torch.__version__)"

如果安裝成功,您將看到PyTorch的版本號。

現在您已經在Ubuntu上安裝了PyTorch,可以開始使用它進行深度學習項目了。

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