是的,CentOS支持PyTorch的GPU版本。要在CentOS上安裝支持GPU的PyTorch,您需要安裝CUDA和cuDNN,然后從PyTorch官方網站下載并安裝適合您CUDA版本的PyTorch二進制文件。以下是具體步驟:
conda install pytorch torchvision torchaudio cudatoolkit11.3 -c pytorch -c conda-forge
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果您看到PyTorch的版本號以及True
(表示CUDA可用),則表示安裝成功。
請注意,以上步驟可能會因系統版本、CUDA版本和PyTorch版本的不同而有所變化。建議查閱PyTorch的官方文檔以獲取最準確的安裝指南。