I am trying to run ROCm's tensorflow port for AMD gpu's. Got the docker container as described here. But I get error when I try to run it

$ drun rocm/tensorflow:latest
docker: Error response from daemon: error gathering device information while adding custom 
device "/dev/kfd": no such file or directory.
ERRO[0000] error waiting for container: context canceled 

Found out that kfd means Kernel fusion driver. But how do I install it?

Some system info -

$ sudo lscpu | grep -i Model
Model:               19
Model name:          AMD A10-6800K APU with Radeon(tm) HD Graphics
$ docker -v
Docker version 19.03.8, build afacb8b7f0
$ uname -r

Running 18.04.4 LTS server

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.