在运行py文件时,经常说缺少各种各样的库,如’ cv2’,‘pytorch’。
首先,检查自己是不是真的安装完了这些依赖包。
可以先试着注释掉 import cv2
,如果注释掉后,引入的其他库仍然报错,有可能是没安装上依赖包,可以到pytorch官网进行安装:Pytorch官网
我安装的是CUDA12.1版本,可以根据自己的CUDA版本进行安装。
使用nvidia-smi
可以查看自己的CUDA版本。
此处给出CUDA12.1的安装指令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
如果注释掉之后正常了,那么应该确实是缺少cv2的库,安装即可:
pip install opencv-python
注意要在对应运行环境下安装
检查cv2是否安装成功:
pip show opencv-python
如果经过以上步骤都没问题,库全部都是正确安装在对应环境上的
那么,应该是环境没有设置正确。
打开pycharm,打开Files-Settings,按照如下图所示进行操作
其中,添加的这个路径是你对应环境下site-packages的位置,我是Anconda环境,因此对应路径就是Anconda/yolov5/Lib/site-packages 如果是Python环境下则选择python中对应环境的site-packages即可。
添加完路径之后,再次编译就有效啦,没生效的话重启一下pycharm或许就行了。
附:
个人blog:Fruit-lxe的blog