在Ubuntu上運行PyTorch,您需要先安裝Python和pip(如果尚未安裝),然后安裝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
例如,如果您想安裝支持CUDA 11.1的PyTorch,可以使用以下命令:
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
如果您不需要GPU支持,可以安裝CPU版本的PyTorch:
pip install torch torchvision torchaudio
python -c "import torch; print(torch.__version__)"
如果安裝成功,您將看到PyTorch的版本號。
現在您已經在Ubuntu上安裝了PyTorch,可以開始使用它進行深度學習項目了。