Hello @jI-mi,
That CML version is kind of old. If I'm not wrong, that version comes with DS 1.5.2.
There the only supported versions for CUDA where 11.4 and 11.1, so you will not be able to do that.
https://docs-archive.cloudera.com/machine-learning/1.5.2/runtimes/topics/ml-runtimes-nvidia-gpu.html
And actually, even the most recent versions only support CUDA 12.5.1
Due to Tensorflow requirements, looks like CUDA 12.8 is not planned to be supported.
You can check with your Cloudera Account Team to request this feature providing your business needs.
Regards, Andrés Fallas -- Was your question answered? Please take some time to click on "Accept as Solution" below this post. If you find a reply useful, say thanks by clicking on the thumbs-up button.