Ubuntu系統中PyTorch的安裝步驟

ubuntu

小樊

2025-08-19 22:47:15

欄目: 智能運維

以下是在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安裝：
- CPU版本：pip install torch torchvision torchaudio
- GPU版本：先安裝對應CUDA Toolkit，再執行pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cuXXX（XXX替換為CUDA版本號，如cu118）
使用conda安裝：
- CPU版本：conda create -n pytorch_env python=3.8，conda activate pytorch_env，conda install pytorch torchvision torchaudio
- GPU版本：conda create -n pytorch_env python=3.8，conda activate pytorch_env，conda install pytorch torchvision torchaudio cudatoolkit=XXX -c pytorch（XXX替換為CUDA版本號）

python -c "import torch; print(torch.__version__)"
python -c "import torch.cuda; print(torch.cuda.is_available())"

最新問答