From 178a64f360348ee46d619edadd6edd15600e0a05 Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Tue, 1 Nov 2022 00:42:29 -0700
Subject: [PATCH] chore: update CUDA_VISIBLE_DEVICES env

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
---
 src/bentoml_cli/env.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/bentoml_cli/env.py b/src/bentoml_cli/env.py
index 369e94d9891..7a41248ebb9 100644
--- 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",
 ]