溫馨提示×

Debian上PyTorch的常用命令有哪些

小樊
49
2025-09-03 11:15:29
欄目: 智能運維

以下是Debian上PyTorch的常用命令,涵蓋安裝、驗證及環境管理:

1. 系統與依賴準備

  • 更新系統包:
    sudo apt update && sudo apt upgrade -y
  • 安裝Python和pip:
    sudo apt install python3 python3-pip

2. PyTorch安裝命令

  • CPU版本(無GPU)
    pip install torch torchvision torchaudio
  • GPU版本(需先安裝CUDA)
    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
    cu117替換為實際CUDA版本,如cu102、cu120等)
  • Conda安裝(推薦隔離環境)
    conda create -n pytorch_env python=3.9 -y  
    conda activate pytorch_env  
    conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch  # 替換CUDA版本  
    

3. 驗證安裝

  • 檢查版本及CUDA可用性:
    python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

4. 環境管理(可選)

  • 創建虛擬環境(隔離項目):
    python3 -m venv pytorch_env
    source pytorch_env/bin/activate
  • 退出虛擬環境:
    deactivate

5. 其他常用操作

  • 卸載PyTorch:
    pip uninstall torch torchvision torchaudio
  • 查看已安裝包:
    pip list

說明:安裝時需根據系統CUDA版本選擇對應命令,可通過nvcc --version查看CUDA版本。推薦使用虛擬環境隔離不同項目依賴。

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