溫馨提示×

Debian系統PyTorch的安裝步驟是什么

小樊
43
2025-09-13 11:56:35
欄目: 智能運維

在Debian系統上安裝PyTorch,可參考以下步驟:

方法一:使用pip安裝

  1. 更新系統包:sudo apt update。
  2. 安裝Python和pip:sudo apt install python3 python3-pip。
  3. (可選)創建虛擬環境:python3 -m venv pytorch_env,然后source pytorch_env/bin/activate。
  4. 根據CUDA版本選擇安裝命令:
    • 無CUDA(CPU版本):pip install torch torchvision torchaudio。
    • 有CUDA(以CUDA 11.7為例):pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。

方法二:使用conda安裝

  1. 安裝Miniconda:從官網下載并安裝。
  2. (可選)創建虛擬環境:conda create -n pytorch_env python=3.8,然后conda activate pytorch_env。
  3. 根據CUDA版本選擇安裝命令:
    • 無CUDA(CPU版本):conda install pytorch torchvision torchaudio cpuonly -c pytorch。
    • 有CUDA(以CUDA 11.7為例):conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch。

安裝完成后,可通過import torch,print(torch.__version__)print(torch.cuda.is_available())驗證安裝是否成功。

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