Import torch in python. I did these multiple .