docker/run_docker.py · 5f37a5a2bac580f9d9f0016ff474a01a7f4fb989 · chenzk / AlphaFold2_jax

No longer use `runtime=nvidia` to expose GPUs to the container. · 5f37a5a2

Tom Ward authored Mar 02, 2022

Instead use `DeviceRequest`, which is equivalent to running with `--gpu all`.

PiperOrigin-RevId: 431933575
Change-Id: Iee9f70ae8f291d16850e4f58349559d9468ef4d8

5f37a5a2

run_docker.py 10.4 KB

Replace run_docker.py