From a7c5dbe9f5f6c915245af5bef85618eab3bc1d21 Mon Sep 17 00:00:00 2001
From: Wiktor Walentynowicz <wiktor.walentynowicz@pwr.edu.pl>
Date: Mon, 9 Jan 2023 11:26:07 +0100
Subject: [PATCH] Changes in GPU Dockerfile.

---
 .gitignore           | 7 ++++++-
 Dockerfile.gpu       | 4 ++--
 requirements-gpu.txt | 5 -----
 requirements.txt     | 2 +-
 4 files changed, 9 insertions(+), 9 deletions(-)
 delete mode 100644 requirements-gpu.txt

diff --git a/.gitignore b/.gitignore
index c0f7607..b811e01 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,9 @@
 __pycache__
 .tox
 /data
-/venv
\ No newline at end of file
+/venv
+
+# Testing configuration
+docker-compose.yml
+test-config.ini
+models/
\ No newline at end of file
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index ff4cde9..d5b2890 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,8 +1,8 @@
-FROM nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu22.04
+FROM nvidia/cuda:11.7.0-base-ubuntu22.04
 
 RUN DEBIAN_FRONTEND=noninteractive apt-get update && apt-get install -y gcc python3-dev python3-venv python3-pip
 
-COPY requirements-gpu.txt requirements.txt
+COPY requirements.txt requirements.txt
 RUN python3 -m pip install -r requirements.txt && rm requirements.txt
 
 WORKDIR /home/worker
diff --git a/requirements-gpu.txt b/requirements-gpu.txt
deleted file mode 100644
index a01e4ca..0000000
--- a/requirements-gpu.txt
+++ /dev/null
@@ -1,5 +0,0 @@
-torch==1.13.0
---index-url https://pypi.clarin-pl.eu/simple/ 
-nlp_ws
-winer==0.4.1
-awscli==1.22.57
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index b9f2ff4..e432719 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 --index-url https://pypi.clarin-pl.eu/simple/ 
 nlp_ws
-winer==0.4.1
+winer==0.4.2
 awscli==1.22.57
\ No newline at end of file
-- 
GitLab