Haicaji
Haicaji
Do you use an mirror image to install the library? If you use a Tsinghua image to install, the PyTorch you installed is a CPU version.
Please go to the official website to download the corresponding version.