add device parameter for torch.cuda.max_memory_allocated

yandexdataschool / Practical_DL

DL course co-developed by YSDA, HSE and Skoltech

MIT License

1.54k stars 633 forks source link

Closed artemxx closed 3 years ago

artemxx commented 3 years ago

without device it returns 0

review-notebook-app[bot] commented 3 years ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.