weixin_44608943的博客

原创 M2使用GPU版本的Pytorch

cuda是适配nvidia的GPU的，M1及以上芯片中的GPU适配的是mps，在Mac对应操作系统中已经具备，无需单独安装。只需要安装适配的pytorch即可。