溫馨提示×

Ubuntu上如何安裝PyTorch的官方教程

小樊
43
2025-06-23 16:52:35
欄目: 智能運維

在Ubuntu上安裝PyTorch的官方教程如下:

安裝前準備

  1. 更新系統包列表
sudo apt update
  1. 安裝Python和pip(如果尚未安裝):
sudo apt install python3 python3-pip
  1. 安裝虛擬環境(可選,但推薦):
sudo apt install python3-venv

安裝PyTorch

  1. 創建一個新的虛擬環境(可選):
python3 -m venv pytorch-env
  1. 激活虛擬環境(如果創建了虛擬環境):
source pytorch-env/bin/activate
  1. 根據您的CUDA版本選擇合適的PyTorch安裝命令。您可以在PyTorch官方網站找到適用于Ubuntu的安裝命令:PyTorch官方安裝命令
  • 無GPU支持(CPU版本)
pip install torch torchvision torchaudio
  • 有GPU支持(例如CUDA 11.1)
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

驗證安裝

在Python中輸入以下代碼來驗證PyTorch是否成功安裝:

import torch
print(torch.__version__)

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

請注意,以上信息僅供參考,具體安裝步驟可能會因Ubuntu版本、Python版本和CUDA版本的不同而有所差異。建議您在安裝前仔細閱讀PyTorch官方文檔,并根據自己的實際情況進行調整。

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