diff --git a/Dockerfile b/Dockerfile
index 83984224..fcec3cde 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -27,9 +27,7 @@ COPY . /usr/src/app
 # sudo docker image prune -af && sudo docker build -t friendlyhello . && sudo docker tag friendlyhello ultralytics/yolov3:v0
 
 # Run container
-# time sudo docker run -it --memory=8g --cpus=4 ultralytics/yolov3:v0 bash -c './run.sh /1047.tif /tmp && cat /tmp/1047.tif.txt'
-
-# time sudo docker run -it --memory=8g --cpus=4 ultralytics/yolov3:v0 bash -c 'python3 detect.py'
+# time sudo nvidia-docker run ultralytics/yolov3:v0 python3 detect.py
 
 # Push container to https://hub.docker.com/u/ultralytics
 # sudo docker push ultralytics/xview:v30
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index efeae3cf..33daa707 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 # pip3 install -U -r requirements.txt
 numpy
 opencv-python
-torch >= 1.1.0
+torch >= 1.2
 matplotlib
 pycocotools
 tqdm