cellDancer does not use GPU

GuangyuWangLab2021 / cellDancer

Predict RNA velocity through deep learning

https://guangyuwanglab2021.github.io/cellDancer_website/

BSD 3-Clause "New" or "Revised" License

60 stars 11 forks source link

cellDancer does not use GPU #28

Open AlexanderAivazidis opened 7 months ago

AlexanderAivazidis commented 7 months ago

Hi! I just saw that cellDancer is not using the GPU on my system (by running the nvidia-smi command). Is that expected? If not is there anything I can try to change it? I installed everything as advised and ran through the tutorial as shown in your documentation. Thank you! Alexander

Abclisy commented 5 months ago

Hello,

Thank you for reaching out. We did not specify the device or benchmark the resource usage in GPU in our algorithm. We use pytorch lighting in the algorithm so you can change the Python script to specify the device.

Best, Shengyu