My GPU is barely under load, but my CPU is at 50% while training

xuebinqin / DIS

This is the repo for our new project Highly Accurate Dichotomous Image Segmentation

Apache License 2.0

2.11k stars 244 forks source link

My GPU is barely under load, but my CPU is at 50% while training #119

Closed pavelhamburh closed 2 months ago

pavelhamburh commented 2 months ago

Is the training supposed to use only the GPU's memory and none of its computational power? Because my CPU is under a lot of stress (about 50%), while my GPU (4090) shows 0% usage; only the memory is being utilized. Or did I install some requirements incorrectly?

pavelhamburh commented 2 months ago

upd: installed CUDA 11.8 and a relevant torch, vent from 30 sec/it to 1 sec/it

see this repo: https://github.com/HUANGYming/DIS-A100-4090