PyTorch在Ubuntu上的依赖如何解决

在Ubuntu上安装PyTorch依赖可参考以下方法：

使用pip安装

• 先更新系统包列表：sudo apt update。
• 安装Python和pip：sudo apt install python3 python3-pip。
• 根据CUDA版本选择安装命令，如无GPU用pip3 install torch torchvision torchaudio；有CUDA则用对应版本命令，如CUDA 11.7用pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117。

使用conda安装

• 安装Miniconda或Anaconda。
• 创建新环境：conda create -n pytorch_env python=3.8。
• 激活环境：conda activate pytorch_env。
• 安装PyTorch，无GPU用conda install pytorch torchvision torchaudio cpuonly -c pytorch；有CUDA则用conda install pytorch torchvision torchaudio cudatoolkit=对应版本 -c pytorch。

安装完成后，可通过import torch和print(torch.__version__)以及print(torch.cuda.is_available())验证安装是否成功。