deeplearning plugin to work with cuda 9.0

Options
longhowlam
longhowlam Partner, Registered Posts: 24 Partner
Hi all,

is there an easy way to get the deeplearing plugin work with CUDA 9.0

https://www.dataiku.com/dss/plugins/info/deeplearning-image.html

I make use of preinstalled image on GCP with CUDA 9.0 installed, I dont want to setup CUDA 8.0

Regards, Longhow

Best Answer

  • Alex_Combessie
    Alex_Combessie Alpha Tester, Dataiker Alumni Posts: 539 ✭✭✭✭✭✭✭✭✭
    Answer ✓
    Options
    Hi,

    This plugin has been developed for CUDA 8.0, using tensorflow 1.4. This version of tensorflow does not support CUDA 9.0.

    We have logged your feature request to update the tensorflow and CUDA versions.

    If you are designing deep learning workflows in Dataiku, I would advise using directly our Deep Learning interface: https://doc.dataiku.com/dss/latest/machine-learning/deep-learning.

    It is a bit more advanced (i.e. code is involved) than the plugin you mention, but it has some advantages:

    - it supports both CUDA 8 and CUDA 9

    - it also supports transfer learning

    - you can use it for any supervised problem: image classification, text sentiment analysis, time series forecasting, etc.

    - any network architecture can be used within the Keras ecosystem

    - it can be exported as a real-time REST API, similarly to our other ML models

    Assuming you have some experience with Keras, here is a great step-by-step tutorial on how to use it for image classification, with all code included: https://www.dataiku.com/learn/guide/visual/machine-learning/deep-learning-images.html

    Hope it helps,

    Alex

Answers

Setup Info
    Tags
      Help me…