Skip to content

Commit 5f8d918

Browse files
committed
Added details on CPU/GPU specific builds requirements
1 parent 4e77aa4 commit 5f8d918

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,12 @@ Similarly, the `tensorflow_opencv` and `cudnn_tensorflow_opencv` tags follow the
110110

111111
## 3. <a name='Buildingtheimages'></a>Building the images
112112

113+
Building a GPU container requires [`nvidia-docker`](https://github.com/NVIDIA/nvidia-docker) and its `nvidia` container runtime to be the default. To do so, after installation, add `"default-runtime": "nvidia"` to the `/etc/docker/daemon.json` configuration file and restarting the docker daemon using `sudo systemctl restart docker` as described in https://docs.nvidia.com/dgx/nvidia-container-runtime-upgrade/index.html#using-nv-container-runtime . You can test it is valid by running `docker run --rm -it nvidia/cuda:11.2.1-runtime nvidia-smi` from the command line (this command does not have the `--runtime nvidia`). If you get details on your GPU(s) setup, `nvidia` is your `Default Runtime` (use `docker info` for further details).
114+
115+
Building a CPU image requires the `nvidia` runtime disabled: comment the `default-runtime` line above and restart docker.
116+
117+
On GPU build: we do our best effort to specify the GPU architectures; please let us know if some that should be present are missing.
118+
113119
The tag for any image built will contain the `datamachines/` organization addition that is found in the publicly released pre-built container images.
114120

115121
Use the provided `Makefile` by running `make` to get a list of targets to build:

0 commit comments

Comments
 (0)