在CentOS上安裝PyTorch的詳細步驟如下:
系統更新: 首先,確保你的CentOS系統是最新的。
sudo yum update -y
安裝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.8 -c pytorch
驗證安裝: 激活虛擬環境后,運行以下Python代碼以驗證PyTorch是否安裝成功:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
通過以上步驟,你應該能夠在CentOS系統上成功安裝PyTorch。