温馨提示×

centos安装pytorch有哪些步骤

小樊
81
2025-01-02 00:33:08
栏目: 智能运维

在CentOS上安装PyTorch通常涉及以下步骤:

  1. 更新系统: 首先,确保你的CentOS系统是最新的。你可以使用以下命令来更新系统:

    sudo yum update -y
    
  2. 安装Miniconda: Miniconda是一个轻量级的Anaconda发行版,包含了conda、Python以及180多个科学包及其依赖项。你可以从Miniconda的官方网站下载适合你系统的安装包,并按照安装向导进行安装。

    下载链接:https://docs.conda.io/en/latest/miniconda.html

    安装示例(以Linux版为例):

    wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
    bash Miniconda3-latest-Linux-x86_64.sh
    

    安装过程中,你需要接受许可协议、选择安装路径等。完成后,重启你的shell或终端。

  3. 激活conda环境: 安装完成后,你需要激活conda环境。你可以创建一个新的环境并激活它,或者直接使用base环境。

    创建新环境并激活:

    conda create -n myenv python=3.8
    conda activate myenv
    

    激活base环境(如果你不想创建新环境):

    conda activate base
    
  4. 安装PyTorch: 在激活的环境中,你可以使用conda或pip来安装PyTorch。由于CentOS可能没有官方的PyTorch包,你可能需要添加PyTorch的官方channel。

    使用conda安装:

    conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c conda-forge
    

    注意:cudatoolkit=11.3中的版本号可能需要根据你的CUDA版本进行调整。你可以通过运行conda info cudatoolkit来查看可用的CUDA版本。

    使用pip安装(如果conda安装失败或不可用):

    pip install torch torchvision torchaudio
    
  5. 验证安装: 安装完成后,你可以验证PyTorch是否安装成功。运行以下Python代码:

    import torch
    print(torch.__version__)
    print(torch.cuda.is_available())
    

    如果一切正常,你应该能看到PyTorch的版本号以及CUDA是否可用(取决于你的系统配置)。

请注意,以上步骤可能会因系统版本、CUDA版本和PyTorch版本的不同而有所变化。建议查阅PyTorch的官方文档以获取最准确的安装指南。

0