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