lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image Python extension:

简单的记录一下

报错：

D:\Program Files (x86)\Anaconda3\envs\segmentation\lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image Python extension:
  warn(f"Failed to load image Python extension: {e}")

解决方法：在这里下载与自己CUDA版本和torch版本对应的安装包（.whl文件）

http://download.pytorch.org/whl/torch_stable.html

 

然后使用

pip install **************.whl

进行安装。

可以更准确的安装GPU版本的torchvision

同时安装pytorch时，也建议从上述连接中下载对应CUDA版本的pytorch，在进行安装。

因为直接使用命令安装很有可能就安装成CPU版本的pytorch。