Using keras with tensorflow-rocm

Machine Learning and Modeling
#1

I'm trying to use an AMD GPU for deep learning and I was able to successfully install tensorflow-rocm using either the tensorflow or keras package installer:

install_keras(tensorflow = "...path/to/tensorflow-rocm-package.whl")

or

install_tensorflow(version = "...path/to/tensorflow-rocm-package.whl")

obtained from https://pypi.org/project/tensorflow-rocm/#files

However, but when I tried to use keras after installation I get the following error:

Error: Installation of TensorFlow not found.

Python environments searched for 'tensorflow' package:
 /home/chevrox/.local/share/r-miniconda/envs/r-reticulate/bin/python3.6

You can install TensorFlow using the install_tensorflow() function.

Help appreciated.

#2

Welcome to the R Studio Community!

All the keras for R docs I find specify you must have an NVIDIA GPU for this package. Do you have any doc references that state AMD GPUs are supported?

#4

R interface to Keras


This website provides documentation for the R interface to Keras. See the main Keras website at https://keras.io for additional information on the project.

There are differences between the R interface and the Python API for Keras.

#5

This topic was automatically closed 21 days after the last reply. New replies are no longer allowed.

#3

Nope, but I had hoped that as long as a rocm backend is specified and installed it would work as usual. After all, keras does not use a customized version of tensorflow, does it? Since tensorflow-rocm has the same syntax as tensorflow, I was hoping that it would work.