RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and in

Rachel
Registered Posts: 4 ✭✭
I use GPU clusters in the backend and want to run the recipe in Cuda but it failed.
Could anyone tell me what should I do?
Job failed: Error in python process: At line 9: <class 'RuntimeError'>: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx More info about this error
Best Answer
-
Alexandru Dataiker, Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Adv Designer, Registered Posts: 1,250 Dataiker
Hi @rachelchen
,
Could you please create a support ticket with job and instance diagnostics so we can understand how you've set up and enabled CUDA?
You can check the docs here :
https://doc.dataiku.com/dss/latest/machine-learning/deep-learning/runtime-gpu.html
If using containerized execution you can check if you properly enabled runtime additions "GPU Support"
Answers
-
Thanks AlexT, It works after setting GPU support for Torch 2!