I installed pytorch 1.7 and cuda 11
then I start learning pytorch but when I implement the code in the following link: https://pytorch.org/tutorials/beginner/blitz/ci
