上海微信网站建设公司电话,深圳商城网站哪家做的好,企业常用系统,网站用的服务器背景#xff1a; 程序中有pytorch模型两个#xff0c;yolov5#xff0c;crnn。
之前无论是pth格式#xff0c;还是TRT格式#xff0c;并行的都没有问题。
最近发现#xff0c;多线程ThreadPoolExecutor(max_workers2)调用的时候#xff0c;即单个进程内处理一张图像 程序中有pytorch模型两个yolov5crnn。
之前无论是pth格式还是TRT格式并行的都没有问题。
最近发现多线程ThreadPoolExecutor(max_workers2)调用的时候即单个进程内处理一张图像依次是yolov5---crnn模型会导致GPU利用100%卡住。 起初怀疑是GPU显卡3090计算能力不够但换成2080Ti后依然卡住。
后来发现卡住的pytorch是2.0之前顺利运行的版本都是1.9.1cu111。
后来把pytorch版本换回啦就好了。
具体换的方式是直接把别的环境里装好的torch对应的库拷贝现在的环境。
拷贝的内容包括torch、torch-1.9.1cu111.dist-info、 torchvision、torchvision-0.10.1cu111.dist-info 。文件所在的位置是miniconda3/envs/XXX/lib/python3.8/site-packages/
