Maybe you should notice the CUDA of  your computer ,which must  be correspond with the version of pytorch! So please check it  and choose the right version of pytorch!  I hope my advise maybe usefull.Good luck!