PyTorch在CentOS上的安装步骤 - 问答

在CentOS上安装PyTorch的步骤如下：

1. 安装Miniconda

首先，下载并安装Miniconda，这是一个轻量级的Anaconda发行版，包含conda、Python以及依赖包。

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
chmod +x Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh

按照提示完成安装。

2. 创建并激活Conda环境

创建一个新的Conda环境，并激活它。假设我们创建一个名为torch的环境，并使用Python 3.10。

conda create -n torch python=3.10
conda activate torch

3. 安装CUDA和cuDNN

根据你的GPU型号和CentOS系统，从NVIDIA官方网站下载并安装合适的CUDA Toolkit和cuDNN库。

安装CUDA Toolkit

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
sudo sh cuda_12.1.1_530.30.02_linux.run

安装cuDNN

wget https://developer.nvidia.com/compute/machine-learning/cudnn/12.1.1.30/Production/12.1_20211031/cudnn-12.1.1.30-linux-x64-v8.0.4.30.tgz
tar -xzvf cudnn-12.1.1.30-linux-x64-v8.0.4.30.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

4. 安装PyTorch

在激活的Conda环境中，使用pip安装PyTorch。你可以从PyTorch官方网站获取安装命令。

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

5. 验证安装

创建一个名为test_pytorch.py的Python脚本，内容如下：

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

运行此脚本以验证PyTorch是否已成功安装：

python test_pytorch.py

如果输出显示PyTorch的版本号和True（表示CUDA可用），则表示安装成功。

以上步骤应该可以帮助你在CentOS上成功安装PyTorch。如果在安装过程中遇到问题，可以参考PyTorch官方文档或相关社区论坛寻求帮助。

0 赞

0 踩