在 AutoDL 租显卡, 创建机器的时候选择 算法镜像——lora-scripts 即可。
我的这个镜像,xformers 是基于 sm86 架构编译的,本身我是在北京区的3090显卡上制作的这个镜像,推荐你也直接使用3090显卡。
也可以尝试使用这些显卡:RTX4090, RTX3080
其他显卡不做正常运行保障。如果你用其他显卡,请自行重新安装xformers。
开机完毕,进入jupyterlab
进入以后应该是这样子的。
(可选操作) 这时候推荐你先点击一下终端,输入`git pull && git submodule update`然后按回车。保持和我的脚本一更新到最新版。
依赖已经完全安装好,什么都不需要动,直接打开train.sh 编辑参数。
修改完毕参数一定要保存。这时候我们打开终端,输入`bash train.sh`即可开始训练。