to_cpu
Browse files
app.py
CHANGED
@@ -346,8 +346,8 @@ class Experiment(Component):
|
|
346 |
# @spaces.GPU
|
347 |
def render_plots(data_id, *metric_inputs):
|
348 |
# Clear Cache Files
|
349 |
-
print(f"GPU Check: {torch.cuda.is_available()}")
|
350 |
-
print("Which GPU: ", torch.cuda.current_device())
|
351 |
cache_dir = f"{os.environ['GRADIO_TEMP_DIR']}/res"
|
352 |
if not os.path.exists(cache_dir): os.makedirs(cache_dir)
|
353 |
for f in os.listdir(cache_dir):
|
|
|
346 |
# @spaces.GPU
|
347 |
def render_plots(data_id, *metric_inputs):
|
348 |
# Clear Cache Files
|
349 |
+
# print(f"GPU Check: {torch.cuda.is_available()}")
|
350 |
+
# print("Which GPU: ", torch.cuda.current_device())
|
351 |
cache_dir = f"{os.environ['GRADIO_TEMP_DIR']}/res"
|
352 |
if not os.path.exists(cache_dir): os.makedirs(cache_dir)
|
353 |
for f in os.listdir(cache_dir):
|