Closed andrewsonin closed 2 years ago
Hi, @andrewsonin. Thank you for your interest in our library.
We are planning to add a separate setup for NLP, CV and reporting (may be others) blocks. However, some will not have PyTorch from which one could easily snatch GPU IDs (as we do now, if you don't pass gpu_ids
). This seems to be solved through parsing aforenamed nvidia-smi
. I think that along with the pull request for a separate installation, we will add the exception you proposed.
Stale issue message
If I have only two available GPUs, I can still pass
gpu_id='2'
to theTabularAutoML
without getting an error.nvidia-smi
output:Code:
The kernel appears to have died at the
fit_predict
stage when it tries to train theCatBoost
model.Changing
gpu_ids='2'
togpu_ids='1'
solves the problem.Please add the corresponding check to the
__init__
of theTabularAutoML
.