在CentOS上安裝PyTorch前,需先確保系統為最新狀態,并安裝核心開發工具與庫,避免后續依賴沖突。
1. 更新系統包
執行以下命令同步系統軟件包及安全補?。?/p>
sudo yum update -y
2. 安裝基礎開發工具
通過Development Tools組安裝gcc、make、cmake等編譯工具(PyTorch部分組件需編譯):
sudo yum groupinstall -y "Development Tools"
3. 安裝Python環境
PyTorch要求Python 3.7及以上版本,通過yum安裝Python3及pip(Python包管理工具):
sudo yum install -y python3 python3-devel python3-pip
4. 創建虛擬環境(可選但推薦)
隔離項目依賴,避免全局環境污染:
python3 -m venv pytorch_env # 創建名為pytorch_env的虛擬環境
source pytorch_env/bin/activate # 激活虛擬環境(后續操作均在激活狀態下進行)
PyTorch的正常運行需依賴數值計算、科學計算等基礎庫,需提前安裝:
sudo yum install -y numpy scipy matplotlib pandas # 數值計算與數據可視化庫
sudo yum install -y blas-devel lapack-devel # 線性代數運算庫(PyTorch底層依賴)
若需使用PyTorch的GPU加速功能,需安裝CUDA Toolkit(NVIDIA GPU驅動支持)和cuDNN(深度神經網絡加速庫),且版本需與PyTorch版本兼容(參考PyTorch官網兼容性表)。
1. 安裝CUDA Toolkit
sudo rpm -i cuda-repo-rhel7-<version>.rpm # 替換<version>為實際版本號
sudo yum clean all
sudo yum install -y cuda
~/.bashrc文件,添加以下內容:export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
運行source ~/.bashrc使配置生效。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*
根據是否需要GPU支持,選擇對應的安裝命令(務必參考PyTorch官網最新命令,避免版本沖突):
1. CPU版本(無GPU加速)
直接通過pip安裝預編譯的CPU版本:
pip install torch torchvision torchaudio
2. GPU版本(需CUDA支持)
使用--extra-index-url指定CUDA版本的PyTorch輪子地址(如CUDA 11.7):
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
提示:若需指定PyTorch版本(如2.0.1),可將命令改為:
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --extra-index-url https://download.pytorch.org/whl/cu117
安裝完成后,通過Python交互環境驗證PyTorch是否安裝成功及GPU是否可用:
import torch
print("PyTorch版本:", torch.__version__) # 輸出PyTorch版本號
print("CUDA是否可用:", torch.cuda.is_available()) # 若為True,說明GPU加速可用
1. 依賴沖突
若安裝過程中出現“依賴沖突”錯誤(如某庫版本不兼容),可使用--no-cache-dir禁用pip緩存,或用yum的--setopt=obsoletes=0忽略過時包:
pip install --no-cache-dir torch torchvision torchaudio
# 或
sudo yum install -y --setopt=obsoletes=0 <package_name>
2. 權限問題
若因權限不足導致安裝失敗,可在命令前添加sudo(不推薦用于虛擬環境),或在虛擬環境中操作(推薦)。
3. 網絡問題(中國大陸)
使用國內鏡像源加速安裝(如清華源),避免因網絡延遲導致失?。?/p>
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
4. GLIBC版本過低
若出現“ImportError: /lib64/libc.so.6: version `GLIBC_2.14’ not found”錯誤,說明系統GLIBC版本過低(PyTorch需GLIBC 2.14及以上)??蓢L試升級GLIBC(需謹慎,可能影響系統穩定性),或安裝與系統GLIBC版本兼容的PyTorch版本(如PyTorch 1.12及以下)。
通過以上步驟,可解決CentOS上PyTorch安裝的大部分依賴問題。若仍有報錯,建議參考PyTorch官方文檔或社區論壇(如Stack Overflow)獲取針對性解決方案。