chore: update CUDA_VISIBLE_DEVICES env

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Nov 1, 2022 · 178a64f · 178a64f
1 parent a808612
commit 178a64f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/bentoml_cli/env.py b/src/bentoml_cli/env.py
@@ -36,6 +36,7 @@
     "RAYON_RS_NUM_CPUS",
     "TF_NUM_INTEROP_THREADS",
     "TF_NUM_INTRAOP_THREADS",
+    "CUDA_VISIBLE_DEVICES",
 ]