单机多卡运行卡死 about medicalgpt

[shibing624]

执行命令是啥？

[zhangxinxin0428]

@shibing624 bash run_pt.sh

[shibing624]

@shibing624 bash run_pt.sh

我在colab上3分钟跑完了这个演示的，你试试。

[zhangxinxin0428]

感谢您的回复，如果您直接跑这个代码没有问题的话，那可能就是我硬件环境的问题了

[zhangxinxin0428]

/### Describe the Question Please provide a clear and concise description of what the question is. 直接运行run_pt.sh后，模型正常加载，到数据那一步卡死了，也不报错，也不往下走，显卡显存也卡住不动

Describe your attempts

• I walked through the tutorials
• I checked the documentation
• I checked to make sure that this is not a duplicate question
• [ ]
  

等待十几分钟后，报错如下：

有没有小伙伴和我遇到同样的问题？想请教一下如何解决

[shibing624]

你在colab上试了pretrain的ok吗？ 如果ok，把本地环境安装的跟colab一样就行（全部库是最新版本）。

[nlper-hou]

这个问题我遇到过，可以把batch_size降低，或者删一部分数据，用少量数据跑，就可以了

[zhangxinxin0428]

这个问题我遇到过，可以把batch_size降低，或者删一部分数据，用少量数据跑，就可以了

好的，我这边试一下，感谢您的解答