scikeras icon indicating copy to clipboard operation
scikeras copied to clipboard
verified publisher icon adriangb

n_jobs increase the memory of GPU

Open talhaanwarch opened this issue 4 years ago • 2 comments

With default n_jobs i am using less than 2GB out of 8GB GPU ram. But as soon as i change n_jobs to 2, my GPU ram goes to 8 GB.

talhaanwarch avatar Nov 30 '21 06:11 talhaanwarch

Is the expectation that it only doubles? Can you provide a minimal working example so that I can reproduce? Thanks!

adriangb avatar Nov 30 '21 07:11 adriangb

n_jobs=1 image n_jobs=2 image

same code as #257

talhaanwarch avatar Nov 30 '21 10:11 talhaanwarch