Docker Community Forums

Share and learn in the Docker community.

GPU access while building

Hi I have a Dockerfile:

FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 
RUN apt update
RUN apt install -y vim
RUN apt install -y python3
RUN apt install -y python3-pip
RUN apt install -y wget
...
<something>
...
RUN python3 make_checkpoint.py

This python script uses tensorflow with a GPU. Else it would take ages. Is it possible to make the GPU accessible while building?