Docker Community Forums

Share and learn in the Docker community.

Tensorflow gpu support on custom docker image

Hello everyone,im new to docker.When trying to install tensorflow-gpu==1.15 through it fails to detect gpu and throws error like could not load dynamic library etc.I have nvidia docker installed and alsouse the gpus all flag.Can some one please help

Turns out it was as simple as installing tensorflow-gpu not tensorflow, odd that the tensorflow docs don’t explain that but basically this was me being dumb.

thanks for your help