溫馨提示×

Ubuntu PyTorch安裝包在哪里下載

小樊
60
2025-08-28 00:47:24
欄目: 智能運維

在Ubuntu上安裝PyTorch,可通過以下兩種常見方式下載安裝包:

  • 使用pip安裝:訪問PyTorch官方網站,根據系統、CUDA版本等信息生成安裝命令,如在有CUDA 11.8支持的情況下,可執行命令pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118來下載安裝。若無GPU,可使用pip3 install torch torchvision torchaudio命令安裝CPU版本。
  • 使用conda安裝:先安裝Miniconda或Anaconda,然后創建虛擬環境并激活。以安裝CUDA 11.8版本的PyTorch為例,可執行命令conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia,無GPU則使用conda install pytorch torchvision torchaudio cpuonly -c pytorch命令。

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