Win10 system installs pytorch-GPU environment, easy to use in personal test

The cuda11.2 version is installed locally on the computer, and the graphics card driver configuration is as follows:

Go to the official website to check and install the pytorch -GPU version command: Previous PyTorch Versions | PyTorch

Select the red box version installation command

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

 installing

 Test whether the installation is successful