PyTorch在CentOS上的安装步骤是什么

在CentOS上安装PyTorch可以通过多种方式实现，以下是使用Anaconda和源码编译两种常见方法的详细步骤：

使用Anaconda安装PyTorch

1. 安装Anaconda：

• 访问Anaconda官网下载并安装适合CentOS的版本。

2. 创建虚拟环境：

• 打开终端，使用以下命令创建一个名为pytorch的虚拟环境，并指定Python版本（例如Python 3.8）：

  conda create -n pytorch python=3.8
  

  复制代码

3. 激活虚拟环境：

• 激活刚刚创建的虚拟环境：

  conda activate pytorch
  

  复制代码

4. 安装PyTorch：

• 在激活的虚拟环境中，使用以下命令安装PyTorch。请根据你的具体需求选择CPU版本或带有CUDA支持的版本：

  conda install pytorch torchvision torchaudio cpuonly -c pytorch
  

  复制代码
  若要安装支持CUDA的版本，请确保你的系统已安装NVIDIA CUDA工具包，并替换cpuonly为cudatoolkit。

5. 验证安装：

• 激活虚拟环境后，输入以下命令验证PyTorch是否安装成功：

  python -c "import torch; print(torch.__version__)"
  

  复制代码
  如果没有报错并输出了PyTorch的版本号，则表示安装成功。

使用源码编译安装PyTorch

1. 安装依赖：

• 确保系统已安装必要的依赖包：

  yum install -y patch libjpeg-turbo-devel dos2unix git
  

  复制代码

2. 下载PyTorch源码：

• 克隆PyTorch的GitHub仓库：

  git clone -b v1.8.1 https://github.com/pytorch/pytorch.git --depth 1 pytorch_v1.8.1
  cd pytorch_v1.8.1
  

  复制代码

3. 配置环境变量：

• 设置环境变量以便编译和安装：

  export USE_XNNPACK=0
  

  复制代码

4. 编译安装：

• 执行编译安装命令：

  python3 setup.py install
  

  复制代码

5. 安装PyTorch插件：

• 安装生成的PyTorch插件包：

  pip3 install --upgrade dist/torch_npu-1.8.1-cp37-cp37m-linux_aarch64.whl
  

  复制代码
  请根据你的系统架构和Python版本选择相应的whl文件。

6. 验证安装：

• 进入测试目录并执行单元测试脚本以验证安装是否成功：

  cd test/test_network_ops/python3
  ./test_div.py
  

  复制代码
  如果结果显示OK，则表示PyTorch安装成功。

请注意，源码编译安装可能需要较长时间，并且需要确保系统具备足够的资源。此外，如果你需要安装支持GPU的PyTorch版本，还需要安装CUDA和cuDNN，并在编译时指定相应的选项。

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