CentOS支持PyTorch的GPU版本,具体如下:
- 支持的GPU型号:主要支持NVIDIA GPU,包括GeForce GTX 10系列、RTX系列、Quadro系列、Tesla系列等,如GTX 1050 Ti、RTX 2080 Ti、Quadro P系列、Tesla V100等。从PyTorch 1.8版本开始,官方加入了对AMD ROCm的支持,理论上AMD GPUs也可用于CentOS系统运行PyTorch,但可能存在一些限制和兼容性问题。
 
- 支持的关键条件:需安装与GPU型号兼容的CUDA Toolkit和cuDNN库,同时要确保PyTorch版本与CUDA版本匹配。例如,PyTorch 1.9支持CUDA 10.2、11.1和11.3等。