在centos上安裝pytorch及其依賴項，可以按照以下步驟進(jìn)行：
1. 更新系統(tǒng)
首先，確保你的centos系統(tǒng)是最新的：
sudo yum update -y
2. 安裝必要的依賴項
安裝構(gòu)建pytorch所需的依賴項：
sudo yum groupinstall -y "Development Tools" sudo yum install -y cmake3 git wget sudo yum install -y Python3 python3-devel python3-pip
3. 安裝CUDA（如果需要GPU支持）
如果你有NVIDIA GPU并且想要使用GPU加速，你需要安裝CUDA。可以從NVIDIA官網(wǎng)下載適合你系統(tǒng)的CUDA Toolkit。
下載CUDA Toolkit
訪問NVIDIA CUDA Toolkit下載頁面，選擇適合你系統(tǒng)的版本并下載。
安裝CUDA Toolkit
假設(shè)你下載的是.rpm文件，可以使用以下命令安裝：
sudo rpm -i cuda-repo-rhel7-<version>.rpm sudo yum clean all sudo yum install -y cuda
配置環(huán)境變量
編輯~/.bashrc文件，添加以下行：
export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
然后運行：
source ~/.bashrc
4. 安裝cuDNN（如果需要GPU支持）
cuDNN是用于深度神經(jīng)網(wǎng)絡(luò)的GPU加速庫。你需要從NVIDIA官網(wǎng)下載適合你CUDA版本的cuDNN。
下載cuDNN
訪問NVIDIA cuDNN下載頁面，選擇適合你CUDA版本的cuDNN并下載。
安裝cuDNN
假設(shè)你下載的是.tgz文件，可以使用以下命令安裝：
tar -xzvf cudnn-<version>-Linux-x64-v<version>.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*
5. 安裝PyTorch
你可以使用pip來安裝PyTorch。首先，確保你已經(jīng)安裝了pip：
sudo yum install -y python3-pip
然后，安裝PyTorch。你可以從PyTorch官網(wǎng)獲取適合你系統(tǒng)的安裝命令。例如，如果你使用的是CPU版本：
pip3 install torch torchvision torchaudio
如果你使用的是GPU版本，并且已經(jīng)安裝了CUDA和cuDNN，可以使用以下命令：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
請根據(jù)你的CUDA版本替換cu113。
6. 驗證安裝
安裝完成后，你可以驗證PyTorch是否安裝成功：
import torch print(torch.__version__) print(torch.cuda.is_available()) # 如果使用GPU版本，應(yīng)該返回True
通過以上步驟，你應(yīng)該能夠在CentOS上成功安裝PyTorch及其依賴項。
