温馨提示×

CentOS支持哪些版本的PyTorch

centos

小樊

2025-08-15 00:13:45

栏目: 智能运维

CentOS支持的PyTorch版本取决于CUDA和cuDNN版本，具体如下：

CentOS 7.6及以上：推荐使用Python 3.6 - 3.9，可安装多个版本的PyTorch，如PyTorch 1.9.0需搭配CUDA 11.1，PyTorch 2.0.1需搭配CUDA 11.3等。
无GPU支持：可安装CPU版本，通过conda install pytorch torchvision torchaudio cpuonly -c pytorch命令安装。

0 赞

0 踩