From 20ff7a19d9d9eb24c8ec7a0d7422228b4c69086e Mon Sep 17 00:00:00 2001
From: meidlinga <meidlinga@users.noreply.github.com>
Date: Sat, 1 Oct 2022 19:09:28 +0200
Subject: [PATCH] chore: update installation instructions

---
 classify_images.py | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/classify_images.py b/classify_images.py
index 5f57685..04e33e8 100644
--- a/classify_images.py
+++ b/classify_images.py
@@ -20,14 +20,26 @@
 #
 # Dependencies (may work against later versions, but this was the test environment):
 #
-# conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
-# pip install pretrainedmodels==0.7.4
-# pip install pillow==6.1.0
-# pip install progressbar2==3.51.0
-# pip install cupy-cuda100==7.3.0
-# pip install torchnet==0.0.4
+# Using conda:
+#   conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=11.7 -c pytorch
+# Alternativly using apt under Ubuntu 22.04:
+#   apt install -y nvidia-cuda-toolkit
+#   pip install cupy-cuda11x
+# Alternativly using apt under Ubuntu 20.04:
+#   apt install -y nvidia-cuda-toolkit
+#   pip install cupy-cuda101
 #
-####### 
+# Then install:
+# pip install \
+#   matplotlib==3.5.1 \
+#   pandas==1.3.5 \
+#   pillow==9.2.0 \
+#   pretrainedmodels==0.7.4 \
+#   progressbar2==3.51.0 \
+#   scikit-image==0.19.3 \
+#   torchnet==0.0.4
+#
+#######
 
 #%% Imports