ubuntu

Ubuntu下PyTorch的依赖项有哪些

小樊
42
2025-08-19 22:45:10
栏目: 智能运维

在Ubuntu下安装PyTorch,若使用pip或conda安装,主要依赖项有Python(推荐3.6及以上版本)、pip、CUDA(可选,用于GPU加速)、cuDNN(可选,与CUDA配套用于GPU加速)。若采用编译安装,除上述依赖项外,还需安装build-essential、cmake、libopenblas-dev、liblapack-dev等开发工具和库。

0
看了该问题的人还看了