The way to let pertpy use GPU

I am using pertpy 0.4.0. My computer has GPU RTX A4000.

Can anyone please help me how I can let pertpy run the GPU?

Thank you very much!

On my computer, tensorflow and pytorch could use the GPU.


pertpy uses Jax for GPU acceleration. Can you verify that you can use the GPU with Jax? I want to add that only scGen can be GPU accelerated within pertpy at the moment.

1 Like

Thank you for helping me. I assumed that this site would sent me a notification to my gmail like github . I am sorry that I did not receive this notification so I did not know your reply earlier.