linux服务器怎么安装pytorch

569
2024/11/30 12:23:29
栏目: 云计算
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Linux服务器上安装PyTorch通常涉及几个步骤,包括安装必要的依赖项、获取PyTorch软件包以及编译和安装。以下是一个基本的指南,假设你使用的是基于Debian的系统(如Ubuntu)或基于RPM的系统(如CentOS)。

基于Debian的系统(如Ubuntu)

  1. 更新包列表

    sudo apt update
    
  2. 安装必要的依赖项

    sudo apt install -y build-essential cmake git libopenblas-dev liblapack-dev libjpeg-dev libpng-dev libtiff-dev libavcodec-dev libavformat-dev libswscale-dev python3-dev python3-pip
    
  3. 安装Python3和pip(如果尚未安装):

    sudo apt install -y python3 python3-pip
    
  4. 创建一个新的虚拟环境(可选)

    python3 -m venv pytorch_env
    source pytorch_env/bin/activate
    
  5. 安装PyTorch: 你可以通过pip安装PyTorch的二进制版本。访问PyTorch官方网站获取最新的安装命令。例如:

    pip3 install torch torchvision torchaudio
    

基于RPM的系统(如CentOS)

  1. 安装必要的依赖项

    sudo yum install -y gcc-c++ cmake git openblas-devel lapack-devel libjpeg-turbo-devel libpng-devel libtiff-devel ffmpeg-devel python3-devel python3-pip
    
  2. 安装Python3和pip(如果尚未安装):

    sudo yum install -y python3 python3-pip
    
  3. 创建一个新的虚拟环境(可选)

    python3 -m venv pytorch_env
    source pytorch_env/bin/activate
    
  4. 安装PyTorch: 同样,你可以通过pip安装PyTorch的二进制版本。访问PyTorch官方网站获取最新的安装命令。例如:

    pip3 install torch torchvision torchaudio
    

使用conda安装(推荐)

如果你使用的是Anaconda或Miniconda,可以通过conda来安装PyTorch,这通常更简单且环境管理更方便。

  1. 安装Miniconda(如果尚未安装): 下载并安装Miniconda:

    wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
    bash Miniconda3-latest-Linux-x86_64.sh
    
  2. 创建一个新的虚拟环境(可选)

    conda create -n pytorch_env python=3.8
    conda activate pytorch_env
    
  3. 安装PyTorch

    conda install pytorch torchvision torchaudio cpuonly -c pytorch
    

    如果你有NVIDIA GPU并且希望使用GPU加速,可以安装带有CUDA的版本:

    conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
    

请根据你的具体需求和环境选择合适的安装方法。如果你遇到任何问题,可以参考PyTorch官方文档获取更多信息。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: linux中rmdir命令的用法是什么