要在Ubuntu中安裝PyTorch,可以使用conda或pip進行安裝。以下是使用conda安裝PyTorch的步驟:
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
source ~/.bashrc
conda create -n myenv
conda activate myenv
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安裝完成后,你就可以在Python腳本中使用PyTorch。以下是一個簡單的示例:
import torch
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])
z = x + y
print(z)
運行上述代碼將輸出兩個張量的和。這表明PyTorch已經成功安裝和使用。
你也可以使用pip來安裝PyTorch。以下是使用pip安裝PyTorch的步驟:
sudo apt update
sudo apt install python3-pip
pip install torch torchvision torchaudio
安裝完成后,你可以按照上述示例使用PyTorch。
希望這可以幫助你在Ubuntu中安裝和使用PyTorch。