From 9394245a7e0fc24ff44a3fb2d2290b6071408b27 Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Tue, 4 Apr 2023 13:41:29 +0000
Subject: [PATCH] update docker

---
 docker/Dockerfile       | 4 ++--
 docker/requirements.txt | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 3060e57..53414d4 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 # Use an existing image as the base
-FROM nvcr.io/nvidia/pytorch:21.10-py3
+FROM nvcr.io/nvidia/pytorch:22.06-py3
 
 # Add the current user to the image
 ARG USER_NAME=lhunor
@@ -8,7 +8,7 @@ ARG GROUP_NAME=guest
 ARG GROUP_ID=11000
 
 # Update the system and install required packages
-RUN apt-get update && apt-get upgrade -y && apt-get install -y sudo git apt-utils
+RUN apt-get update && apt-get upgrade -y && apt-get install -y sudo git apt-utils htop
 
 # Create a new group with the same GID as the host
 RUN groupadd --gid $GROUP_ID $GROUP_NAME
diff --git a/docker/requirements.txt b/docker/requirements.txt
index 2f3990c..33ff92b 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -7,4 +7,6 @@ plotly
 pandas
 cmapy
 pytorch3d
-tables
\ No newline at end of file
+tables
+numpy-stl
+torchmetrics
\ No newline at end of file