Add ncnn auto thread count (#3847)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
single_channel
nihui 1 year ago committed by GitHub
parent b251b7415a
commit 01dcd54b19
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -3,7 +3,6 @@
import ast import ast
import contextlib import contextlib
import json import json
import os
import platform import platform
import zipfile import zipfile
from collections import OrderedDict, namedtuple from collections import OrderedDict, namedtuple
@ -267,7 +266,6 @@ class AutoBackend(nn.Module):
check_requirements('git+https://github.com/Tencent/ncnn.git' if ARM64 else 'ncnn') # requires ncnn check_requirements('git+https://github.com/Tencent/ncnn.git' if ARM64 else 'ncnn') # requires ncnn
import ncnn as pyncnn import ncnn as pyncnn
net = pyncnn.Net() net = pyncnn.Net()
net.opt.num_threads = os.cpu_count()
net.opt.use_vulkan_compute = cuda net.opt.use_vulkan_compute = cuda
w = Path(w) w = Path(w)
if not w.is_file(): # if not *.param if not w.is_file(): # if not *.param

Loading…
Cancel
Save