PtyLab / PtyLab.py

Python implementation of ptylab
Other
18 stars 8 forks source link

Setter checks GPU availability when manually set #62

Closed ShantanuKodgirwar closed 4 months ago

ShantanuKodgirwar commented 4 months ago

@roflmaostc As I mentioned in this closed PR https://github.com/PtyLab/PtyLab.py/pull/61, after one manually sets attribute params.gpuSwitch, the availability of cupy and cuda should perhaps be tested. I have implemented these things in this small PR and I'll summarize the 4 conditions that are implemented :

dboonz commented 4 months ago

Hi Shantanu, thank you for your great work! Apologies for the slowness, I'll merge it now.