在CentOS上安裝PyTorch的步驟如下:
conda --version
,若顯示版本號,則安裝成功。創建虛擬環境命令:
conda create -n pytorch_env python=3.8
Python 3.8是兼容PyTorch的穩定版本,可替換為3.9或3.10(需確認PyTorch支持)。
激活環境:
conda activate pytorch_env
激活后終端提示符應顯示 (pytorch_env)
。
CPU版本:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
GPU版本(需NVIDIA顯卡和CUDA 11.8):
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
啟動Python解釋器:
python
在Python解釋器中輸入以下代碼驗證PyTorch安裝:
import torch
x = torch.rand(5, 3)
print(x)
如果輸出類似于 tensor([[0.3089, 0.5494, 0.1153], [0.9371, 0.8981, 0.3798], [0.3220, 0.4789, 0.6379], [0.3280, 0.0454, 0.5896], [0.2760, 0.0586, 0.2910]])
的結果,則表示PyTorch安裝成功。
如果安裝速度慢或失敗,可以使用國內鏡像源(如清華源):
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
然后使用以下命令安裝PyTorch:
pip install torch torchvision torchaudio
以上步驟應該可以幫助你在CentOS上成功安裝PyTorch。如果在安裝過程中遇到問題,可以參考PyTorch官方文檔或相關社區論壇尋求幫助。