溫馨提示×

如何在Ubuntu上運行PyTorch示例

小樊
42
2025-07-01 19:59:37
欄目: 智能運維

在Ubuntu上運行PyTorch示例,你需要按照以下步驟操作:

  1. 安裝Python和pip: Ubuntu通常預裝了Python,但可能不是最新版本。你可以通過以下命令安裝Python 3和pip(如果尚未安裝):

    sudo apt update
    sudo apt install python3 python3-pip
    
  2. 安裝PyTorch: PyTorch官方網站提供了針對不同操作系統和CUDA版本的PyTorch安裝指令。訪問PyTorch官網(https://pytorch.org/get-started/locally/),選擇適合你的Ubuntu版本和CUDA版本的安裝命令。例如,如果你想使用CPU版本的PyTorch,可以使用以下命令:

    pip3 install torch torchvision torchaudio
    

    如果你想使用GPU版本的PyTorch,并且你的系統有兼容的NVIDIA GPU以及已安裝CUDA,可以使用類似下面的命令:

    pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
    

    請根據你的CUDA版本選擇正確的URL。

  3. 驗證安裝: 安裝完成后,你可以通過運行Python解釋器并嘗試導入PyTorch來驗證安裝是否成功:

    python3
    

    然后在Python解釋器中輸入:

    import torch
    print(torch.__version__)
    

    如果沒有錯誤信息,并且打印出了PyTorch的版本號,說明安裝成功。

  4. 運行PyTorch示例: PyTorch官方網站提供了許多示例代碼,你可以從這里找到它們:https://pytorch.org/tutorials/

    選擇一個示例,將其代碼保存到一個.py文件中,例如example.py。然后使用Python運行這個腳本:

    python3 example.py
    

    這將執行腳本中的代碼,你應該能夠看到示例的輸出。

請注意,根據你的具體需求(例如,特定的PyTorch版本或CUDA版本),安裝步驟可能會有所不同。始終參考PyTorch官方網站上的最新指南來獲取最準確的安裝指令。

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