GPU support for recipes
![bkos_vis](https://us.v-cdn.net/6038231/uploads/Dataiku/nAvatar3.png)
Hi,
Is it possible to run code recipes with GPU? I am able to train object detection or classification tasks on the GPU. However it seems that when I run custom code recipes or notebooks, the GPU isn't used.
Answers
-
Sergey Dataiker, Dataiku DSS Core Designer, Dataiku DSS & SQL, Dataiku DSS Core Concepts Posts: 365 Dataiker
Hi @bkos_vis
If you are talking about R or Python recipes, the answer to your question is based truly on the code you are running. DSS doesn't change anything in the code execution and runs it "as is". In other words, if the code supposed to use GPU, it should use it.
Why are thinking GPU is not used? Are you using "nvidia-smi -l" to verify?
-
You are right, there was a compatibility issue between my CUDA and TensorFlow version that didn't show up in the logs. Therefore I wanted to know if DSS shielded off some resources during execution.
Thank you for your answer!