在CentOS上安裝PyTorch需要以下依賴:
具體安裝步驟如下:
安裝Anaconda3: 從Anaconda官網下載并安裝適合CentOS系統的Anaconda3安裝包。
創建虛擬環境: 使用conda創建一個新的虛擬環境,例如:
conda create -n pytorch python=3.8
conda activate pytorch
安裝PyTorch: 在激活的虛擬環境中,使用conda安裝PyTorch。如果需要GPU支持,確保已安裝相應版本的CUDA和cuDNN,并選擇支持GPU的版本:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
請注意,cudatoolkit
的版本號可能需要根據你的CUDA版本進行調整。你可以通過運行conda info cudatoolkit
來查看可用的CUDA版本。
驗證安裝: 安裝完成后,可以通過運行以下Python代碼來驗證PyTorch是否安裝成功,并且能夠使用GPU:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果一切正常,你應該能看到PyTorch的版本號以及CUDA是否可用(取決于你的系統配置)。