要在Ubuntu上安裝PyTorch并使用它,您可以按照以下步驟進行操作:
安裝Anaconda:在終端中執行以下命令,下載并安裝Anaconda,這是一個用于管理Python環境的工具。
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
在安裝過程中,按照提示進行配置。安裝完成后,執行source ~/.bashrc
命令,使Anaconda生效。
創建新的Anaconda環境:在終端中執行以下命令,創建一個名為pytorch的新環境,并激活該環境。
conda create -n pytorch python=3.8
conda activate pytorch
安裝PyTorch:在激活的pytorch環境中,執行以下命令,安裝PyTorch。
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
驗證安裝:在Python交互式環境中,執行以下命令,驗證PyTorch是否安裝成功。
import torch
print(torch.__version__)
如果能夠成功輸出PyTorch的版本號,則表示安裝成功。
現在,您可以使用PyTorch進行深度學習任務了。可以查閱PyTorch的官方文檔來學習如何使用PyTorch的各種功能和API。