made so code can be trained on GPU

• update readme to install environment

Func/DrawImages.py

@@ -6,7 +6,7 @@ def plot_images(images, labels, class_indices):
     num_images = len(images)
     grid_size = int(np.ceil(np.sqrt(num_images)))
 
-    plt.figure(figsize=(15, 15))
+    plt.figure()
     for i in range(num_images):
         plt.subplot(grid_size, grid_size, i + 1)
         img = images[i]