溫馨提示×

Ubuntu PyTorch如何配置環境

小樊
49
2025-08-19 22:54:15
欄目: 智能運維

以下是在Ubuntu上配置PyTorch環境的步驟:

安裝Python和pip(若未安裝)

sudo apt update
sudo apt install python3 python3-pip

創建虛擬環境(可選但推薦)

python3 -m venv pytorch_env
source pytorch_env/bin/activate

安裝PyTorch

  • 使用pip安裝
    • CPU版本pip3 install torch torchvision torchaudio
    • GPU版本(以CUDA 11.8為例)pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
  • 使用conda安裝
    • CPU版本conda install pytorch torchvision torchaudio cpuonly -c pytorch
    • GPU版本(以CUDA 11.8為例)conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia

驗證安裝

python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

若安裝成功,會顯示PyTorch版本號,GPU版本會顯示True。

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