Ubuntu中PyTorch依赖怎么解决

ubuntu

小樊

2025-09-15 01:40:44

栏目: 智能运维

在Ubuntu中解决PyTorch依赖问题，可参考以下步骤：

安装系统依赖

更新系统包列表：sudo apt update，然后安装必要的构建工具和库：sudo apt install -y build-essential cmake git wget unzip yasm pkg-config libopenblas-dev liblapack-dev libjpeg-dev libpng-dev。

安装Python和pip

使用命令：sudo apt install -y python3 python3-pip。

安装CUDA和cuDNN（可选，用于GPU加速）

从NVIDIA官网下载对应版本的CUDA Toolkit和cuDNN库，然后按照官方文档进行安装和配置，包括设置环境变量等。

安装PyTorch

可以使用pip或conda安装。以pip为例，根据CUDA版本选择合适命令，如CPU版本：pip3 install torch torchvision torchaudio；CUDA 11.7版本：pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。若使用conda，先创建并激活虚拟环境，再执行类似conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch的命令。

验证安装

通过运行import torch，然后执行print(torch.__version__)和print(torch.cuda.is_available())来验证，若能正确显示版本号且GPU版本返回True，则安装成功。