溫馨提示×

ubuntu安裝pytorch怎么使用

小億
91
2024-01-27 10:14:50
欄目: 編程語言

要在Ubuntu上安裝PyTorch并使用它,您可以按照以下步驟進行操作:

  1. 安裝Anaconda:在終端中執(zhí)行以下命令,下載并安裝Anaconda,這是一個用于管理Python環(huán)境的工具。

    wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
    chmod +x Anaconda3-2021.05-Linux-x86_64.sh
    ./Anaconda3-2021.05-Linux-x86_64.sh
    

    在安裝過程中,按照提示進行配置。安裝完成后,執(zhí)行source ~/.bashrc命令,使Anaconda生效。

  2. 創(chuàng)建新的Anaconda環(huán)境:在終端中執(zhí)行以下命令,創(chuàng)建一個名為pytorch的新環(huán)境,并激活該環(huán)境。

    conda create -n pytorch python=3.8
    conda activate pytorch
    
  3. 安裝PyTorch:在激活的pytorch環(huán)境中,執(zhí)行以下命令,安裝PyTorch。

    conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
    
  4. 驗證安裝:在Python交互式環(huán)境中,執(zhí)行以下命令,驗證PyTorch是否安裝成功。

    import torch
    print(torch.__version__)
    

    如果能夠成功輸出PyTorch的版本號,則表示安裝成功。

現(xiàn)在,您可以使用PyTorch進行深度學習任務了。可以查閱PyTorch的官方文檔來學習如何使用PyTorch的各種功能和API。

0