Hi @mrshenli,
I am using Pytorch 1.4.0 installed by the following command on the official website:

image1029×390 20.1 KB

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

Sure, I will check the posts and see if I can figure out the problem by myself. Please let me know if the version is a problem, and it is much appreciated if you could share more thoughts on the code I provided above.
Thank you!
Best,
Ziyi