Installation using Docker

Install Docker

Make sure docker has been installed. You can refer to the official tutorial.

Install NVIDIA Container Toolkit if using GPU. You can refer to nvidia-docker

Docker image

You can build image locally or using pre-build images, pre-built Docker images are available on DockerHub.

Pull Docker Image

You can directly pull the pre-build docker images for athena. We have created the following docker images:

athena-tf-2.1.0-gpu-py3

Download the image as below:

docker pull garygao99/athena:tf-2.1.0-gpu-py3

Build images

You can build image locally as below:

docker build -t garygao99/athena:tf-2.1.0-gpu-py3 .

Create Container

After the image downloaded, create a container.

docker run -it --gpus all garygao99/athena:tf-2.1.0-gpu-py3 /bin/bash