在CentOS上安裝PyTorch的依賴項,通常推薦使用conda來管理,因為它可以自動處理依賴關系并確保環境的一致性。以下是使用conda安裝PyTorch依賴項的步驟:
conda create -n torch_env python=3.8
conda activate torch_env
# 安裝PyTorch和torchvision,指定CUDA版本(以CUDA 11.3為例)
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
import torch
print(torch.__version__)
print(torch.cuda.is_available())
請注意,具體的依賴項和版本可能會隨著PyTorch的更新而變化,因此建議參考PyTorch官方文檔或GitHub頁面上的最新安裝指南來獲取最準確的信息。