centos安装pytorch软件源

要在 CentOS 上安装 PyTorch，您可以使用以下步骤：

1. 首先，确保您的系统已更新并安装了 Python 和 pip。如果没有，请运行以下命令：

sudo yum update
sudo yum install python3 python3-pip

2. 安装 Conda（如果尚未安装），这是一个 Python 包管理器，可以简化安装和管理 Python 软件包的过程。要安装 Miniconda（一个较小的 Anaconda 发行版），请访问 https://docs.conda.io/en/latest/miniconda.html 并按照说明进行操作。

3. 使用 Conda 创建一个新的 Python 环境（可选，但推荐）：

conda create -n torch_env python=3.8
conda activate torch_env

4. 使用 Conda 安装 PyTorch。您可以根据需要选择 CPU 或 GPU 版本。要安装支持 CPU 的 PyTorch，请运行：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

要安装支持 GPU 的 PyTorch，请运行：

conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch

请将 <your_cuda_version> 替换为您的系统上安装的 CUDA 版本，例如 cudatoolkit=11.1。

5. 安装完成后，您可以通过运行以下命令来验证 PyTorch 是否已成功安装：

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

这将输出已安装的 PyTorch 版本。现在您可以在 CentOS 上使用 PyTorch 了！