CentOS上PyTorch版本選擇指南

在CentOS上安裝PyTorch時，選擇正確的版本非常重要，以確保兼容性和性能。以下是一個詳細的指南，幫助你在CentOS上選擇并安裝合適的PyTorch版本。

1. 系統要求

• 操作系統版本: CentOS 7.6或更高版本。
• CUDA版本: 根據你的CUDA版本選擇相應的PyTorch版本。例如，PyTorch 1.9.0需要CUDA 11.1，而PyTorch 2.0.1則需要CUDA 11.3。
• Python版本: PyTorch官方支持Python 3.6-3.9。

2. 安裝Anaconda3

首先，需要在CentOS上安裝Anaconda3，因為PyTorch通常通過conda進行安裝?？梢詮腁naconda官網下載對應版本的安裝包，并按照提示完成安裝。

wget https://repo.anaconda.com/archive/Anaconda3-2024.05-Linux-x86_64.sh
bash Anaconda3-2024.05-Linux-x86_64.sh

3. 創建虛擬環境

建議創建一個虛擬環境來安裝PyTorch，以避免與系統其他部分的潛在沖突。

conda create -n pytorch python=3.8
conda activate pytorch

4. 安裝PyTorch

通過conda安裝PyTorch，選擇與你的CUDA版本相匹配的PyTorch版本。

安裝支持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版本，例如cudatoolkit11.1。

5. 驗證安裝

安裝完成后，可以通過運行以下Python代碼來驗證PyTorch是否正確安裝，并且能夠使用GPU。

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

如果輸出為True，則說明PyTorch已正確安裝并可以使用GPU。

6. 使用清華源加速安裝

為了加快下載速度，可以使用清華大學的鏡像源。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

7. 常見問題及解決方法

• Numpy和torch版本不匹配: 降級Numpy版本。

pip install "numpy<2.0"

• 找不到conda環境: 確保Anaconda已正確安裝，并檢查環境變量。

參考鏈接

• 在Linux服務器上配置Pytorch
• Linux下安裝Pytorch，有了這個教程輕松搞定！
• 極簡版 | pytorch安裝、環境配置及其各種問題解決辦法（適用新版本pycharm+anaconda）
• centos安裝pytorch版本要求 - 問答 - 億速云
• centos安裝pytorch軟件源 - 問答 - 億速云
• centos安裝pytorch兼容版本 - 問答 - 億速云
• centos 7 配置pytorch運行環境 - 谷子弟 - 博客園

通過以上步驟，你應該能夠在CentOS系統上成功安裝并配置PyTorch，開始你的深度學習項目。如果在安裝過程中遇到問題，建議查閱PyTorch官方文檔或尋求社區的幫助。