在centos上安裝pytorch及其相關依賴，可以遵循以下步驟進行操作：
第一步：更新系統
首先，確保你的centos系統是最新的版本：
sudo yum update -y
第二步：安裝必要的依賴
接下來，安裝一些基本的開發工具和庫：
sudo yum groupinstall -y "Development Tools" sudo yum install -y cmake3 git wget
第三步：安裝Python和pip
pytorch需要Python環境，你可以使用Anaconda來管理Python環境和安裝PyTorch：
# 下載并安裝Miniconda wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh bash Miniconda3-latest-Linux-x86_64.sh # 按照提示完成安裝 source ~/.bashrc # 創建一個新的conda環境 conda create -n pytorch_env python=3.8 conda activate pytorch_env
第四步：安裝PyTorch
你可以使用conda來安裝PyTorch，這樣可以確保所有依賴都正確安裝：
# 安裝PyTorch（根據你的CUDA版本選擇合適的命令） conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
如果你不需要GPU支持，可以選擇CPU版本：
conda install pytorch torchvision torchaudio cpuonly -c pytorch
第五步：驗證安裝
安裝完成后，你可以通過以下命令驗證PyTorch是否安裝成功：
import torch print(torch.__version__) print(torch.cuda.is_available()) # 如果安裝了GPU版本，應該返回True
第六步：安裝其他依賴（可選）
根據你的項目需求，可能還需要安裝其他Python庫，可以使用pip來安裝：
pip install numpy pandas matplotlib scipy scikit-learn
總結
通過以上步驟，你應該能夠在CentOS上成功安裝PyTorch及其依賴。如果你遇到任何問題，可以參考PyTorch官方文檔或社區論壇尋求幫助。
