在用YOLOV5训练模型时,遇到如下报错:
Error: /home/hwy/.conda/lib/python3.8/site-packages/torch/lib/../../nvidia/cudnn/lib/libcudnn_ops_train.so.8。我所用的虚拟环境是/home/hwy/.conda。运行代码在虚拟环境中找cudnn,而该虚拟环境中没有。
查看所有虚拟环境
conda env list
查看结果:
用虚拟环境torch,激活虚拟环境torch
source activate torch
顺利训练,不再报错。