search

XiaoTaoWang / EagleC

A deep-learning framework for predicting a full range of structural variations from bulk and single-cell contact maps
Other
52 stars 8 forks source link

specify gpu number to use #25

Open Xieeeee opened 1 year ago

Xieeeee commented 1 year ago

Hi Xiaotao, Thanks for this wonderful tool. I am currently running EagleC on a machine with 4 Quadro RTX 5000 (I hope this would be faster than running on cpu only...) But with one of the gpu largely occupied by another job. However when initiating EagleC run it encounter an error: could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR I assume this could be caused by the low memory in that occupied gpu. I there a way like specifying threads, in specifying number (or which) gpu to use?

Best, Yang

XiaoTaoWang commented 1 year ago

Hi Yang, sorry for the delayed response. Just wanted to let you know that while EagleC used GPU for training, it does not currently support GPU for prediction.