Like a lot of people, i prototype things in notebooks. When it comes to working with GPUs, a common problem i've had to deal with over and over is keeping my environment in delicate balance.
A Stable arrangement I ended up using:
- Ubuntu 18.04
- Docker 19.03
- nvidia-docker https://github.com/NVIDIA/nvidia-docker
- Running notebooks in containers instead of locally, keeping my files and data mounted in volumes.