fixes #40 fixes #41

f0cker · Dec 20, 2023 · 500ceff · 500ceff
1 parent 675a5b6
commit 500ceff
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/docker-compose.nvidia.yml b/docker-compose.nvidia.yml
@@ -16,6 +16,13 @@ services:
             context: ./build
             dockerfile: Dockerfile
         image: "nvidia-ubuntu"
+        deploy:
+                resources:
+                        reservations:
+                                devices:
+                                        -driver: nvidia
+                                        capabilities: [gpu]
+                                        count: all
         ports:
             - "127.0.0.1:8080:8080"
         depends_on:
@@ -28,7 +35,6 @@ services:
             - /var/crackq/:/var/crackq
             - ./crackq:/opt/crackq/build/crackq/
         stdin_open: true
-        runtime: nvidia
         user: crackq
         tty: true
         environment:

diff --git a/docker/common/requirements.txt b/docker/common/requirements.txt
@@ -12,7 +12,7 @@ Flask-SQLAlchemy==3.0.5
 SQLAlchemy-Utils==0.41.1
 flask-talisman==0.7.0
 pysaml2==7.3.1
-Flask-Login==0.6.2
+Flask-Login==0.6.3
 Flask-Cors==3.0.9
 Flask-SeaSurf==1.1.1
 Flask-Migrate==3.0.1

diff --git a/docker/nvidia/ubuntu/Dockerfile b/docker/nvidia/ubuntu/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.0-runtime-ubuntu20.04
+FROM nvidia/cuda:12.3.1-devel-ubuntu20.04
 
 ENV TZ=Europe/London
 ENV DEBIAN_FRONTEND noninteractive