背景
在CPU类型的服务器即使给到足够的运算资源,与GPU类型的服务器做运算来讲仍然是相差甚远,而本人有一台闲置的AMD vega8集显的电脑。想要用来做计算,来探究其与CPU运算的差别。
跟踪
网上查阅资料发现,大部分的用户都是安装NVIDA自家研究的CUDA,然后用作机器学习人工智能方面的运算。而很少有AMD vega显卡的实例。再深度查阅资料发现WSL(Windows子系统linux)中宣布支持vega显卡的并行运算。
开坑
这次准备买一个大规模的硬盘再对Windows系统安装linux子系统挂载硬盘和显卡,自己用作大规模计算和Python相关的学习。
实践
通过网上教程和官方文档,安装好WSL的Ubuntu-20.04分发版,其中比较坑的点:
- 需要禁用旧版控制台:在命令提示符窗口的标签位置右键-属性-选项-取消使用旧版控制台并重启;
- 开启Windows10中的开启linux子系统功能;
- 开启window虚拟化功能和BIOS虚拟化功能;
更新到WSL2才能使用GPU运算;- GPU运算有很多途径,查阅资料发现有:
- NVIDIA GPU 的用途更广,包括但不限于:利用官方驱动安装CUDA在window或linux环境进行机器学习、利用WSL2的驱动使用TensorFlow
- AMD GPU 我查阅了更多资料,发现也有很多途径但都要依赖于官方是否给予相关驱动支持,包括但不限于:ROCm、Direct ML