diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index 7a15cfb4..db433f29 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -14,7 +14,7 @@ RUN rm Dockerfile.cpu
 
 RUN rm Dockerfile.gpu
 
-RUN rm Docker-compose.yml
+RUN rm docker-compose.yml
 
 RUN make
 
@@ -28,7 +28,7 @@ RUN apt-get install -y sudo libgomp1
 
 RUN useradd -U -m yolo
 
-RUN usermod -aG sudo yolo 
+RUN usermod -aG sudo yolo
 
 RUN usermod --shell /bin/bash yolo
 
@@ -45,5 +45,3 @@ RUN ldconfig
 WORKDIR /home/yolo/darknet
 
 USER yolo
-
-
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index c4a9effb..f1985fbc 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -14,11 +14,11 @@ RUN rm Dockerfile.cpu
 
 RUN rm Dockerfile.gpu
 
-RUN rm Docker-compose.yml
+RUN rm docker-compose.yml
 
 RUN make
 
-FROM nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04
 
 ENV DEBIAN_FRONTEND noninteractive
 
@@ -28,7 +28,7 @@ RUN apt-get install -y sudo libgomp1
 
 RUN useradd -U -m yolo
 
-RUN usermod -aG sudo yolo 
+RUN usermod -aG sudo yolo
 
 RUN usermod --shell /bin/bash yolo
 
@@ -44,4 +44,4 @@ RUN ldconfig
 
 WORKDIR /home/yolo/darknet
 
-USER yolo
\ No newline at end of file
+USER yolo
diff --git a/docker-compose.yml b/docker-compose.yml
index 968fbcca..2eebd876 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,7 +2,7 @@ version: '2'
 
 services:
   yolo-gpu:
-    build: 
+    build:
       context: .
       dockerfile: Dockerfile.gpu
     image: yolo:gpu
@@ -14,7 +14,7 @@ services:
               count: 1
               capabilities: [gpu]
   yolo-cpu:
-    build: 
+    build:
       context: .
       dockerfile: Dockerfile.cpu
-    image: yolo:cpu
\ No newline at end of file
+    image: yolo:cpu