I have a laptop with a GeForce 940 MX. I want to get Tensorflow up and running on the gpu. I installed everything from their tutorial page, now when I import Tensorflow, I get>
In case anyone still encounters this. First make sure to add the --runtime=nvidia parameter in order to run your container.
--runtime=nvidia
docker run --runtime=nvidia -t tensorflow/serving:latest-gpu
where tensorflow/serving:latest-gpu is the name of the docker image.
tensorflow/serving:latest-gpu