diff --git a/Dockerfile b/Dockerfile
index 2c29fec..c506a3e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@
 FROM nvcr.io/nvidia/pytorch:20.03-py3
 
 # Install dependencies (pip or conda)
-RUN pip install -U gsutil thop
+RUN pip install -U gsutil
 # RUN pip install -U -r requirements.txt
 
 # Create working directory