CCU:GPU Cluster
Contents
Overview
The CCU provides access to state-of-the-art hardware infrastructure to run GPU-accelerated machine learning applications. This page gives a general overview and links to more in-depth tutorials on how to work with the cluster. There is some overhead involved when writing code for your projects and you have to stick to a few guidelines, but there are template projects and scripts provided so that you can get started with minimal knowledge about the technical background of the GPU cluster.
The GPU cluster is based on Kubernetes, which is a framework to deploy so-called Docker containers to different compute nodes. You can think of a Docker container as a wrapper for your machine learning application, which contains all necessary code and all the libraries it depends on (yes, also the ones from the basic OS). In essence, it is an independent object which can be deployed and run on an arbitrary computer on which the docker infrastructure is installed.
This means for you that you have to be able to take the necessary steps to wrap your own code into a container. All this is covered in an easy, introductory way in the short tutorials below, which should be sufficient to get you started. At some point, you might want to learn about docker in a more in-depth manner, for this, I refer you to the excellent tutorials available elsewhere, some of which are linked here.
What you need
- An account for the CCU.
- Ideally, a desktop PC with an nVidia GPU to test your code before pushing it to the cluster. However, you can develop for and control the cluster on any machine, it's not mandatory that you can actually run the code locally. Note, however, that it makes debugging harder if you cannot do this (you have to do everything on the console).
- Your PC ideally runs a flavor of Linux, all example scripts were tested against Ubuntu 18.04 (should also work on derivatives, such as Mint 19). If you use Windows, you are on your own.
- Admin access to your own PC to install lots of stuff (or a friendly administrator).
- More specific needs will be covered in the in-depth tutorials.
How to get started
- Preparing your system
- Step 1: Install nVidia CUDA and GPU drivers
- Step 2: Install the nVidia docker system
- Step 3: Link to container registry on our server
- For the impatient: Complete install script for a fresh Ubuntu 18.04
- Learning the basics of Docker
- An in-depth look at a container which trains MNIST using Tensorflow, with the following steps:
- Step 1: create a local python tensorflow application.
- Step 2: wrap the application in a container.
- Step 3: run and test the container locally.
- Step 4: push the container to the registry server of the cluster.
- Step 5: remarks on persistent storage in docker containers
- An in-depth look at a container which trains MNIST using Tensorflow, with the following steps:
- Learning the basics of Kubernetes and how to run jobs on the cluster:
- Step 1: Install the Kubernetes infrastructure
- Step 2: Set up your Kubernetes user account
- Step 3: Run the example container on the cluster and make sure that it works correctly.
- Step 4: Persistent volumes on the GPU cluster