楼主可以试试看amd集成显卡跑directML pytorch,这个能比cpu快一些
Cubesat
1. 公式行内显示(inline):请使用 $....$ 或 \(....\) 包裹代码
2. 公式独占一行显示(display):请使用 $$....$$ 或 \[....\] 包裹代码
3. 插入的公式在编辑时不会渲染,请检查无误后再插入。