Pytorch安装出现OSError: [WinError 126] 找不到指定的模块。 Error loading “xx\lib\site-packages\torch\lib\asmjit.dl

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版本

posted @ 2021-12-30 23:49  攻城狮小Liu  阅读(4027)  评论(0)    收藏  举报