/docker

Docker images for the b-it-bots team

Primary LanguageShell

Build Status

b-it-bots Docker images

You can easily create all images using docker-compose:

docker-compose build

GPU Notebook

This image contains several packages for deep learning projects with NVidia GPU support.

  • Build this image

    bash build_and_publish.sh --cuda 11.3.1-cudnn8-runtime-ubuntu20.04 --registry ghcr.io --publish ""

    You can build this image using different cuda versions available here.

  • Run the image locally

    docker run --gpus all --name gpu-notebook -it --rm -d -p 8880:8888 ghcr.io/b-it-bots/docker/gpu-notebook:11.3.1-cudnn8-runtime-ubuntu20.04

  • Login to the container

    docker exec -ti gpu-notebook bash

# check nvidia
nvidia-smi