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