Tensorflow-GPU 2.1 - Recipes to build a container with CUDA 10.1 and CUDNN 7

Prerequisites

To build and execute the container correctly, you need to install the following software:

• NVIDIA driver.
  • If you use ubuntu, follow these instructions to install or update NVIDIA drivers.
• Docker and docker-compose.
• Nvidia-docker2.

---

Building and runing container

On docker-compose.yml, according to your preferences, change the username in args and add the desired volumes.

• To build: make build
• To run: make run
• To execute interactively bash onto the container: make exec