Different memory allocation on GTX 1080 ti, Tesla k80, Tesla v100 for the same pytorch model

后端 未结 0 1815
难免孤独
难免孤独 2021-01-01 08:10

I have tried loading a distilbert model in pytorch over 3 different GPUs (GeForce GTX 1080 ti, tesla k80, tesla v100). According to the pytorch cuda profiler, the memory co

相关标签:
回答
  • 消灭零回复
提交回复
热议问题