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.