diff options
Diffstat (limited to 'docker-compose.yml')
-rw-r--r-- | docker-compose.yml | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/docker-compose.yml b/docker-compose.yml index 1acf735..79df23f 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -21,6 +21,7 @@ services: - HTTP_PROXY=socks5://172.17.0.1:9995 - HTTPS_PROXY=socks5://172.17.0.1:9995 - NO_PROXY="localhost, 127.0.0.1" + - NVIDIA_VISIBLE_DEVICES=all runtime: nvidia volumes: devijlab-vault: |