简单的记录一下
报错:
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。