conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
The installation can be completed normally, but an error occurred in the import torch vision:
C:\Users\xuyic\anaconda3\envs\pytorch\lib\site-packages\torchvision\io\image.py:11: UserWarning: Failed to load image Python extension: [WinError 126] 找不到指定的模块。 warn(f"Failed to load image Python extension: {e}")
Solution:
1. From cuda10.2 to 11.3
2. Reduce the version of torchvision, such as 1.3.0