Installing Tensorflow 2.0 on Ubuntu 18.04 using docker. Run all experiments from a container.

[...\stream_executor\dso_loader.cc] Couldn't open CUDA library nvcuda.dll[...\stream_executor\cuda\cuda_dnn.cc] Unable to load cuDNN DSO

Enter Docker:

Installing Docker and NVIDIA Docker :

Installing tensorflow using Docker:

Fig 1: Output of nvidia-smi inside docker container
docker run -it -rm --runtime=nvidia --name=tensorflow_container tensorflow_image_name
docker pull tensorflow/tensorflow:latest-gpu-py3
Fig 2: Custom dockerfile with tensorflow

Engineering at NAYAN Technologies. On a quest for technology.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store