如何解决在MacBook Pro上使用pytorch Cuda
我正在使用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?
解决方法
否。
CUDA仅适用于受支持的NVidia GPU,不适用于AMD GPU。
一直在努力通过PyTorch(通过ROCm,在MacOS上不起作用)支持AMD GPU的加速。
,CUDA是nVidia为nVidia GPU开发的GPU计算框架。同样,CuDNN框架也是如此。
目前,您不能将PyTorch与AMD GPU一起使用GPU加速,即没有nVidia GPU。 O.S.不是问题,也就是说,您拥有macOC并不重要。这取决于您拥有的GPU。
但是,您可以做的是,您可以购买外部nVidia GPU或使用某些群集。例如,Google Colab提供了PyTorch兼容性。
,遗憾的是 CUDA 仅适用于受支持的 NVidia GPU
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。