Updated dependencies

2024-11-08 01:18:31 +01:00
parent 8c7f2c1c97
commit 512a21cdb3
7 changed files with 12 additions and 12 deletions
--- a/python/Dockerfile
+++ b/python/Dockerfile
@ -1,6 +1,6 @@
-FROM nvidia/cuda:12.5.1-devel-ubi9 AS builder
+FROM nvidia/cuda:12.6.2-devel-ubi9

-RUN dnf install -y python3.12-3.12.1-4.el9 \
+RUN dnf install -y python3.12-3.12.1-4.el9_4.4 \
 	&& dnf clean all \
 	&& ln -s /usr/bin/python3.12 /usr/bin/python

--- a/python/docker-compose.yaml
+++ b/python/docker-compose.yaml
@ -12,5 +12,4 @@ services:
        reservations:
          devices:
            - driver: nvidia
-              count: 1
              capabilities: [gpu]
--- a/python/requirements.txt
+++ b/python/requirements.txt
@ -1,3 +1,5 @@
 numba==0.60.0
-scikit-learn==1.5.1
-tqdm==4.66.4
+scikit-learn==1.5.2
+tqdm==4.67.0
+
+#pudb==2024.1.3