Docker gpu fix (#89)

* GPU support

* check gpu

* get gpu kwargs based on its availability
This commit is contained in:
you-n-g
2024-07-19 16:20:07 +08:00
committed by GitHub
parent 4f558b8203
commit ea8c62a9fb
2 changed files with 20 additions and 0 deletions
+1
View File
@@ -43,6 +43,7 @@ class EnvUtils(unittest.TestCase):
"""
qtde = QTDockerEnv()
qtde.prepare() # you can prepare for multiple times. It is expected to handle it correctly
# qtde.run("nvidia-smi") # NOTE: you can check your GPU with this command
# the stdout are returned as result
result = qtde.run(local_path=str(DIRNAME / "env_tpl"), entry="qrun conf.yaml")