Closed ORCAaAaA-ui closed 2 months ago
Thank you for raising this issue. We did observe the similar issue. Unfortunately, we don't have the direct control the GPU memory in this case. In the solvent models, we use a lot of grids to make sure the accuracy. You can reduce the lebedev_order in
https://github.com/pyscf/gpu4pyscf/blob/master/examples/16-smd_solvent.py#L33
base on
https://github.com/pyscf/pyscf/blob/master/pyscf/dft/LebedevGrid.py#L4999
mf.with_solvent.lebedev_order = 17
should be enough in most cases. This will reduce the memory consumption a lot.
You can also try to call mempool.free_all_blocks()
to release GPU memory before the expensive calculations.
In the future, this part of python code will be moved into cuda kernel for memory efficiency.
The memory usage has been improved in https://github.com/pyscf/gpu4pyscf/pull/150 It would be very helpful if you could test it on your side. You will need to compile the master branch for testing.
Hello,
I am encountering a memory overflow issue while performing geometric optimization on the Tamoxifen molecule with the gpu4pyscf in SMD solvation model. The GPU memory usage momentarily reached its maximum capacity, causing the calculation to terminate. The calculation proceeds without errors for smaller molecules, but fails for Tamoxifen with the following error message:
Has anyone experienced similar issues, or does anyone have suggestions on how to manage memory usage more effectively in such calculations? Any advice or adjustments to settings that might help bypass this memory constraint would be greatly appreciated.
Thank you!