安装windows 下 pytorch时,出现了以上问题。
解决方法:
装一个这个即可解决
no module named cv2
解决方法:
pip install -i https://pypi.doubanio.com/simple python-opencv出错
根据你python的版本,自己下载与python版本对应的opencv-python,网址如下
https://pypi.tuna.tsinghua.edu.cn/simple/opencv-python/
然后打开cmd,找到文件位置 pip install opencv_python-3.2.0.6-cp36-cp36m-win_amd64.whl
附: windows下pytorch安装:pip install torch===1.1.0 torchvision===0.3.0 -i https://pypi.doubanio.com/simple python3.6
最后检测一下是否是安装的gpu版本的pytorch:
print(torch.cuda.is_available())