参考:
The MAX78000 Microcontroller, Some Coffee, and Artificial Intelligence
https://github.com/MaximIntegratedAI/ai8x-training

项目开发路线:

image.png

训练生成的文件分布

  1. ..../ai8x-training/
  2. ..../ai8x-synthesis/
  3. ..../ai8x-synthesis/sdk/

省略号是项目的根目录

准备

PyTorch
不能使用虚拟机,只能安装适用于 Linux 的 Windows 子系统 (WSL2)。操作如下:
https://github.com/MaximIntegratedAI/ai8x-synthesis/blob/develop/docs/WSL2.md

WSL2

适用于 Linux 2 的 Windows 子系统允许在 Ubuntu Linux 下进行模型训练,并具有 CUDA 硬件加速功能,与主要在 Linux 上开发的大量机器学习工具完全兼容。

空间大小

确保有足够的可用空间用于WSL2安装和MAX78000/MAX78002工具以及数据集。强烈建议至少 32 GB 的可用空间,但要求因所用数据集的大小而异。
New versions of Windows (Windows 10 21H2 or newer, and Windows 11) support WSL2, the Windows Subsystem for Windows, with CUDA hardware acceleration.
image.png
好像太老了版本。怎么办
突然发现实验室电脑的版本正好合适,我也找了一块GT 710的显卡装了进去
image.pngimage.png

安装显卡驱动程序

image.png
image.png
完犊子,这个型号的老显卡,没有CUDA 硬件加速的驱动程序。。。。
image.png
把下载类型,改成游戏就好了。。。。服了这个老六
image.png
开始安装:
image.png
完犊子,驱动都安装失败了
image.png
先跳过这一步,往后面进行

确保CUDA可用于视窗

打开命令提示符并运行nvidia-smi
image.png
好像可以啊,只不过版本好像很低,继续往下

WSL2 安装

以管理员权限打开命令提示符并安装 WSL2(注意一定要用管理员身份运行哦~具体办法自己搜索)

  1. wsl --install

image.png
安装完成了!下一步,重启电脑!
image.png
重启之后,就这样了
image.png

在window上使用Ubuntu

整一个用户名和密码
image.png
搞定
image.png
测试一下,可不可以再WSL2用CUDA nvidia-smi
image.png
不能用,绝了。
驱动刚刚还好好的,能么就不能用了
image.png
没办法了,更新驱动吧,又是漫长的等待~(谢谢360驱动管家,助我一臂之力!)