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

Solved!
rachelchen
Level 1
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and in

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

 

 

0 Kudos
1 Solution
AlexT
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"

Screenshot 2024-03-12 at 2.06.49 PM.png

View solution in original post

2 Replies
AlexT
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"

Screenshot 2024-03-12 at 2.06.49 PM.png

rachelchen
Level 1
Author

Thanks AlexT,  It works after setting GPU support for Torch 2!

0 Kudos