conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
可以正常安装完成,但是import torchvision出错:
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}")
根本原因:
Torchvision版本与CUDA不兼容
解决方法:
1. 从CUDA10.2换到11.3 (有效,✔)
2. 降低torchvision版本