CentOS支持PyTorch的GPU版本，需安裝NVIDIA驅動、CUDA Toolkit和cuDNN庫，然后通過pip或conda安裝對應CUDA版本的PyTorch即可。具體步驟如下：
PATH和LD_LIBRARY_PATH）。--extra-index-url https://download.pytorch.org/whl/cu117對應CUDA 11.7）。import torch; print(torch.cuda.is_available())，若輸出True則表示GPU支持已正確配置。
