背景:
程序中有pytorch模型两个,yolov5,crnn。
之前无论是pth格式,还是TRT格式,并行的都没有问题。
最近发现,多线程ThreadPoolExecutor(max_workers=2)调用的时候,即单个进程内处理一张图像,依次是yolov5--->crnn模型,会导致GPU利用100%,卡住。
起初怀疑是:GPU显卡3090计算能力不够,但换成2080Ti后,依然卡住。
后来发现:卡住的pytorch是2.0,之前顺利运行的版本都是1.9.1+cu111。
后来把pytorch版本换回啦就好了。
具体换的方式是:直接把别的环境里装好的torch对应的库,拷贝现在的环境。
拷贝的内容包括torch、torch-1.9.1+cu111.dist-info、 torchvision、torchvision-0.10.1+cu111.dist-info 。文件所在的位置是:miniconda3/envs/XXX/lib/python3.8/site-packages/