Macbook外置显卡和CUDA配置(用于深度学习)
因为平时用Mac比较多,但是深度学习需要Nvidia显卡加速计算,而二货苹果用的A卡不支持,只能用外置Nvidia显卡,最终选择通过雷电3接口外接显卡扩展坞的方式进行配置。然而事实证明这是一条比较艰辛的道路。。。。下面把配置中的相关东西写一下,为各位需要的同学提供参考。
一、硬件配置
1、笔者电脑为Macbook Pro 2018系统信息如下:
2、外置显卡为NVIDIA GeForce GTX 1080,为技嘉 G1 GAMING版本:
3、显卡扩展坞采用雷蛇战核X:
二、主要参考资料
主要参考材料为下面的一篇知乎帖子:
三、注意事项
按照帖子里的步骤一步一步来不会有大的问题,主要有几个要注意的地方。
1、教程针对雷电3接口,老一点的macbook需要看一下是雷电3还是雷电2,雷电2应该也有相应的解决方法,但是效果可能差一些。
顺便吐槽一下:雷3长0.5米的线就要两百多,这完全让人产生了在玩音响器材的错觉。。。。。
2、Macbook连接外置显卡主要的技术支持来自下面的网站:
有很多内容大家可以自己探索一下,包括论坛中各种问题的解决方法。
3、笔者电脑系统当前相对较新,一些东西和原贴内容不一致,出现了一些问题。建议大家安装好了以后请轻易不要升级电脑系统。
4、注意显卡驱动与macOS版本的对应关系。主要注意安装完成之后自己显卡驱动的版本,如果按照推荐使用下面的帖子进行显卡驱动的安装。
一般安装完成后在屏幕的右上角会有一个Nvidia的标志点击打开后在Updates栏可以看到驱动的版本,笔者版本为387.10.10.40.105。
也可以在“系统偏好设置”中找到“NVIDIA Driver Manager”最下面一排查看(如果还没安装CUDA驱动,就没有下面的“CUDA”图标)。
安装哪个驱动版本是安装程序自己默认的,这个不需要操心,知道安装了哪个版本号的驱动就行。
5、在使用script进行驱动安装时,请使用terminal而不是iTem。
6、注意CUDA版本与显卡驱动以及macOS版本的对应关系。具体下载安装哪个版本的CUDA请参考下面的Nvidia官方网页,根据自己的macOS版本和显卡驱动版本进行选择。这一点上很多教程比较含糊,请大家特别注意!
7、关闭Macbook自己的独立显卡!关闭Macbook自己的独立显卡 !关闭Macbook自己的独立显卡 !重要的事情说三遍!笔者被一打开某些特定软件电脑就变卡折腾得心力憔悴,最后发现是需要将笔记本自带的独立显卡关闭,关闭的方法为使用下面的命令:
sudo pmset -a gpuswitch 0
其中参数0表示只使用笔记本的集成显卡iGPU,使用参数1表示只使用自带独立显卡dGPU,使用参数2表示显卡自动切换。 方法参考下面的帖子:
看得出来是一个中国哥们儿写的。You are a real lifesaver!
所以啊,不管干啥行业,学好英语很重要。推荐自己的一个不正经学英语的公众号:【隔壁老王学英语】。
8、启动时存在花屏的问题,由于笔者的电脑从来都是乱造的,如果你在意,请自行寻求解决方法。
9、推荐采用外接屏幕方式,不推荐只使用笔记本内屏。
四、最终效果
最后搭建好的系统如下:
对比进行风格迁移实验(参见笔者下面的知乎帖)
当不使用外接显卡时,使用pytorch框架进行风格迁移计算,一张图片计算时间在15分钟左右,笔记本风扇跟烧水壶一样响 。成功驱动显卡并使用CUDA之后,一张图片计算时间约在半分钟左右,效果显著。
笔者想说的是:苹果不用N卡真的坑啊 !!!!!有这功夫还是自己搞个服务器靠谱。
最后,再腆着老脸推荐一下自己的【隔壁老王学英语】。各位老铁,您都看到这里了,就赏脸关注一个呗。
不想关注?没有关系,先看看老王的英语学习理念,您觉得有道理再来。