建e网室内设计广州seo技术优化网站seo

背景：

      程序中有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/