Questions about CUDA and cuDNN versions for Deep learning extensions.

KHKKHK Member Posts: 7 Contributor I
edited January 2020 in Help
Hi, RapidMiner.   

First of all, thank you very much for making such a great operator.

But I have a problem using the Deep learning extension.

The protocol using the deep learning operator of the Deep learning extension works well when the deep learning backend is set to the CPU, but when the deep learning backend is set to the GPU, it rarely uses the GPU. We also found that the computational speed was also slower than when we set backend with the CPU.

My graphics card is GTX 1080 Ti, CUDA version is 9.0.176, cuDNN version is 7.0. We have also set the environment variables for cuDNN.

Have I missed anything?  
I know you are busy, but I need help. :'-(


Thank you.

Kim.

Best Answers

Answers

  • KHKKHK Member Posts: 7 Contributor I
    edited February 2020
    Hi @David_A

    Sorry for the late response.

    Here is the nvidia-smi screen when the process is in progress.

    And the GPU screen in Task Manager.


    The graphics driver version is the minimum version installed automatically when install CUDA 9.0.

    The same problem occurs when upgrade this graphics driver to the latest version.

    The batch size is 40 and the number of data in the training set is about 4300.



Sign In or Register to comment.