Hello, I've recently switched computers to be able to run the Resnet101 part of the Mask RCNN code. The problem I've been facing is that the program keeps using the CPU instead of the GPU. Current Computer Setup: NVIDIA RTX 3090 24GB GPU ...
const struct cpumask *affinity_mask) { return tags->cpumask; int queue = -1, cpu = 0;set->mq_map = kzalloc_node(sizeof(*set->mq_map) * nr_cpu_ids, GFP_KERNEL, set->numa_node); if (!set->mq_map) return -ENOMEM;if