昨天还能正常实验,今天就突然报错了,到网上找解决方案,大多数都说是版本问题。

于是我怀疑是师兄把cuda版本换了,于是检查pytorch和cuda的版本,发现没的问题。

于是输入指令nvidia-smi查看GPU状态,发现cuda:0的利用率是100%了。

于是将device换成cuda:1,成功运行。

Logo

加入「COC·上海城市开发者社区」,成就更好的自己!

更多推荐