Update for Pytorch container 20.03-py3?

The current pytorch container, 20.03-py3 has some serious bugs including in that version LSTM’s don’t work properly in DataParallel. Is there any plans to update this container soon?

1 Like