• 有的博客说是什么torch版本问题,但是根本不存在,你安装的torch的cuda版本一定是低于机器版本的。所以就是两个地方需要改进:
  • GPUS_PER_NODE=2 #有几张卡就是写几
    batch_size调小

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐