Hi, have you figured this out? I’m kind of doing similar things.
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Does ADVI in PyMC3 support CUDA acceleration? | 0 | 337 | June 7, 2021 | |
| With all the changes, what are the options for ADVI training on GPUs? | 6 | 1168 | July 4, 2022 | |
| Leveraging the GPU in PyMC3 | 5 | 8052 | August 12, 2017 | |
| Can PyMC3 (or even PyMC4) make use of the AMD Radeon "Navi" 5500M GPU in a Macbook Pro 16? | 2 | 1138 | March 25, 2020 | |
| PyMC3 Pickle Issue with GPU | 3 | 2585 | July 24, 2018 |