Solved by zahra in post #9 
              
              
                Thanks. 
Are you running some custom CUDA extensions? No, I am using https://colab.research.google.com/ 
did you build PyTorch from source? No, I am using its modules and functions. 
which GPU are you using? I am using https://colab.research.google.com/ and I do not know what is it. 
I found what ca…