GPU support for recipes

Options
bkos_vis
bkos_vis Registered Posts: 2 ✭✭✭

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.

Tagged:

Answers

  • Sergey
    Sergey Dataiker, Dataiku DSS Core Designer, Dataiku DSS & SQL, Dataiku DSS Core Concepts Posts: 366 Dataiker
    Options

    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?

  • bkos_vis
    bkos_vis Registered Posts: 2 ✭✭✭
    Options

    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!

Setup Info
    Tags
      Help me…