Hello,
I am facing the same problem and tried all the options discussed here, But did not work. Can somebody please help me to solve this problem.
OS: linux
pytorchVesion: 1.0.1
Python version : 3.7.3
Cuda : 9.0.176
Driver vesion: 396.24
Thank you