fix gpu visibility on docker (#98)

jperez999 · web-flow · commit 2aeb862805fb · 2023-02-08T10:31:57.000-08:00
diff --git a/tox.ini b/tox.ini
@@ -64,6 +64,7 @@ commands =
 [testenv:test-gpu]
 passenv =
     NR_USER
+    CUDA_VISIBLE_DEVICES
 sitepackages=true
 ; Runs in: Internal Jenkins
 ; Runs GPU-based tests.
