ubuntu

Ubuntu与PyTorch的集成方式

小樊
36
2025-07-22 00:56:10
栏目: 智能运维

在Ubuntu系统上集成PyTorch可以通过多种方式实现,主要包括使用pip或Anaconda进行安装,并选择是否支持GPU加速。以下是详细的步骤:

使用pip安装PyTorch

  1. 更新系统软件包
sudo apt update
sudo apt upgrade
  1. 安装Python3和pip(如果尚未安装):
sudo apt install -y python3 python3-pip
  1. 创建并激活虚拟环境(可选但推荐):
python3 -m venv pytorch_env
source pytorch_env/bin/activate
  1. 安装PyTorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

请将118替换为你的CUDA版本。

  1. 验证安装
python -c "import torch; print(torch.__version__)"

使用Anaconda安装PyTorch

  1. 安装Anaconda
wget https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh
bash Anaconda3-2023.07-2-Linux-x86_64.sh
source ~/.bashrc
  1. 创建并激活新的conda环境
conda create -n pytorch_env python=3.8
conda activate pytorch_env
  1. 安装PyTorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

请将11.7替换为你的CUDA版本。

  1. 验证安装
python -c "import torch; print(torch.cuda.is_available())"

使用GPU版本的注意事项

通过以上步骤,你应该能够在Ubuntu系统上成功集成PyTorch,无论是用于CPU还是GPU加速的深度学习任务。

0
看了该问题的人还看了