这是因为gpu版本的pytorch自带一个其他版本的cuda。我们还是需要在windows上去英伟达官网下载安装对应显卡驱动的cuda。否则训练效果很差。

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐