The way to let pertpy use GPU

Help
1

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.

2

Hi,

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
3

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.