Commit 2ac03155 authored by Thomas Lips's avatar Thomas Lips
Browse files

update readme and add pytest

parent 1dfe1581
Pipeline #52723 passed with stage
in 9 minutes and 56 seconds
**.pem
\ No newline at end of file
ARG UBUNTU_VERSION=20.04
ARG CUDA = 10.2
FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}.1-base-ubuntu${UBUNTU_VERSION} as base
......@@ -11,3 +11,4 @@ wandb
ipywidgets
tqdm
pre-commit
pytest
......@@ -15,6 +15,8 @@ Docker resources for GPULab
- clone this repo as a submodule in the `.devcontainer` folder
- configure the `.devcontainer.json` to build the desired docker file
- alternatively, use 'image': 'gitlab.ilabt.imec.be:4567/tlips/gpulab/jupyterlab-pytorch:latest' to avoid the need for submodules if you do not want to be able to alter the container. (no auth?)
To enable GPU on local devices: install the [nvidia docker toolkit](https://github.com/NVIDIA/nvidia-docker) and pass the `--gpus all` arg to docker to allow for discovering the GPUs. The nvidia toolkit will use the on-device drivers while installing the cuda toolkit in the container. To validate if the gpus are enabled, run the `nvidia-smi` command in the container.
## IDLab JupyterHub:
- use `gitlab+deploy-token-8:YHsqg1MHW-C4GJpr_B-D@gitlab.ilabt.imec.be:4567/tlips/gpulab/jupyterlab-pytorch:latest` as custom docker image.
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment