Thank you @seemethere , but isnt this the official pytorch docker build image. can you tell me list of things i could change to make it work. Its also not working on nvidia dgx and my friend had the same troubles