fix gpu visibilty issues on privileged container (#1759)

NVIDIA-Merlin · Feb 8, 2023 · 2b19866 · 2b19866
1 parent 93e63d0
commit 2b19866
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tox.ini b/tox.ini
@@ -29,6 +29,7 @@ setenv =
     TF_GPU_ALLOCATOR=cuda_malloc_async
 passenv =
     OPAL_PREFIX NR_USER
+    CUDA_VISIBLE_DEVICES
 sitepackages=true
 ; Runs in: Internal Jenkins
 ; Runs GPU-based tests.