OK That’s great thanks. BTW may I know what was causing it, Because I tried to install pytorch from source in the container and it also didn’t work. However I installed a nightly pytroch binary and it worked.
For my purposes I actually need to make some changes in pytorch source and use that version. But this issue prevents me to do that. Is there any way?