i created a new virtual environment with conda and installed pytorch via source and still am getting the same error. I ran the same code in Titan V (2 gpu and 12gb each) and i am able to execute with no issues.