在Ubuntu上安装PyTorch依赖,可参考以下步骤:
sudo apt update && sudo apt upgrade -y
,确保软件包列表最新。sudo apt install python3 python3-dev python3-pip -y
,安装Python 3及其开发工具和pip。python3 -m venv myenv
创建名为myenv
的虚拟环境,然后通过source myenv/bin/activate
激活它,以隔离项目依赖。pip install torch torchvision torchaudio
;GPU版本需先安装CUDA和cuDNN,再执行pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
(以CUDA 11.7为例)。安装完成后,可通过import torch
和print(torch.__version__)
验证是否安装成功。