path: root/code/sunlab/environment/base/cuda.py

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "1"
from . import *