在MacBook Pro上使用pytorch Cuda

时间:2020-08-15 07:00:17

标签: macos pytorch

我正在使用MacBook Pro(16英寸,2019,macOS 10.15.5(19F96))

GPU

  • AMD Radeon Pro 5300M
  • 英特尔UHD图形630

我正在Mac上尝试将Pytorch与Cuda一起使用。

我看到的所有指南均假定我具有Nvidia图形卡。

我发现了这个问题:https://github.com/pytorch/pytorch/issues/10657,但是看起来我需要安装ROCm,并且根据他们的Supported Operating Systems,它仅支持Linux。

是否可以使用Mac和AMD图形卡在GPU上运行Pytorch?

3 个答案:

答案 0 :(得分:1)

否。

CUDA仅适用于受支持的NVidia GPU,不适用于AMD GPU。

一直在努力通过PyTorch(通过ROCm,在MacOS上不起作用)支持AMD GPU的加速。

答案 1 :(得分:1)

CUDA是nVidia为nVidia GPU开发的GPU计算框架。同样,CuDNN框架也是如此。

目前,您不能将PyTorch与AMD GPU一起使用GPU加速,即没有nVidia GPU。 O.S.不是问题,也就是说,您拥有macOC并不重要。这取决于您拥有的GPU。

但是,您可以做的是,您可以购买外部nVidia GPU或使用某些群集。例如,Google Colab提供了PyTorch兼容性。

答案 2 :(得分:-1)

遗憾的是 CUDA 仅适用于受支持的 NVidia GPU