diff --git a/main.py b/main.py
index f63500f..bfe6b82 100644
--- a/main.py
+++ b/main.py
@@ -1,10 +1,8 @@
 import importlib
-import os
 import sys
 import inspect
 from pathlib import Path
 
-import pynvml
 import hydra
 from omegaconf import DictConfig