在Ubuntu上安裝PyTorch的官方教程如下:
sudo apt update
sudo apt install python3 python3-pip
sudo apt install python3-venv
python3 -m venv pytorch-env
source pytorch-env/bin/activate
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
在Python中輸入以下代碼來驗證PyTorch是否成功安裝:
import torch
print(torch.__version__)
如果安裝成功,您將看到PyTorch的版本號。
請注意,以上信息僅供參考,具體安裝步驟可能會因Ubuntu版本、Python版本和CUDA版本的不同而有所差異。建議您在安裝前仔細閱讀PyTorch官方文檔,并根據自己的實際情況進行調整。