Hello,

I am a student from Professor Jing He's lab at Old Dominion University. I am trying to enable GPU acceleration for our deep learning plugin, DeepSSETracer.

The problem I am running into is that python in the chimerax environment cannot see cuda.

Varying cuda versions seems to rule out including cuda as a dependency. Additionally, we would like to avoid custom cuda installations via the chimerax terminal, as it would greatly complicate installation for the users of our plugin.

Is there a way to support cuda or GPU acceleration in another way (that ideally does not require user interaction with the chimerax terminal)? Or, could you point me towards tools that have managed to do this?

I found this library, ONNX Runtime and wonder if it may be useful: https://onnxruntime.ai/docs/get-started/with-python.html#install-onnx-runtime-gpu-cuda-12x

Best regards,

Bryan Hawickhorst

Old Dominion University