我正在使用MacBook Pro(16英寸,2019,macOS 10.15.5(19F96))
GPU
我正在Mac上尝试将Pytorch与Cuda一起使用。
我看到的所有指南均假定我具有Nvidia图形卡。
我发现了这个问题:https://github.com/pytorch/pytorch/issues/10657,但是看起来我需要安装ROCm,并且根据他们的Supported Operating Systems,它仅支持Linux。
是否可以使用Mac和AMD图形卡在GPU上运行Pytorch?
答案 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