From b85ee9d64a536937912544c7bbd5b98b635b7e8d Mon Sep 17 00:00:00 2001
From: Christian C <cc@localhost>
Date: Mon, 11 Nov 2024 12:29:32 -0800
Subject: Initial commit

---
 code/sunlab/environment/base/cuda.py | 4 ++++
 1 file changed, 4 insertions(+)
 create mode 100644 code/sunlab/environment/base/cuda.py

(limited to 'code/sunlab/environment/base/cuda.py')

diff --git a/code/sunlab/environment/base/cuda.py b/code/sunlab/environment/base/cuda.py
new file mode 100644
index 0000000..a5c813e
--- /dev/null
+++ b/code/sunlab/environment/base/cuda.py
@@ -0,0 +1,4 @@
+import os
+
+os.environ["CUDA_VISIBLE_DEVICES"] = "1"
+from . import *
-- 
cgit v1.2.1