ThomasBlumet
commited on
Commit
·
1dd0e8c
1
Parent(s):
4b4390c
return to normal config without cuda
Browse files- Dockerfile +10 -10
- requirements.txt +3 -4
Dockerfile
CHANGED
@@ -1,13 +1,10 @@
|
|
1 |
# For more information, please refer to https://aka.ms/vscode-docker-python
|
2 |
-
|
3 |
-
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
4 |
|
5 |
-
#nvidia/cuda:
|
6 |
-
|
7 |
-
#ENV DEBIAN_FRONTEND=noninteractive
|
8 |
|
9 |
# Install Python and pip
|
10 |
-
RUN apt-get update && apt-get install -y python3-pip python3-venv
|
11 |
|
12 |
# Where we'll copy the code
|
13 |
WORKDIR /code
|
@@ -16,12 +13,15 @@ WORKDIR /code
|
|
16 |
COPY ./requirements.txt /code/requirements.txt
|
17 |
|
18 |
# Install pip requirements without venv
|
19 |
-
|
20 |
|
21 |
# Create a virtual environment and install pip requirements
|
22 |
-
RUN python3 -m venv /code/venv
|
23 |
-
RUN /code/venv/bin/pip install --no-cache-dir --upgrade pip
|
24 |
-
RUN /code/venv/bin/pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
|
|
|
|
|
|
25 |
|
26 |
# Creates a non-root user with an explicit UID
|
27 |
# For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers
|
|
|
1 |
# For more information, please refer to https://aka.ms/vscode-docker-python
|
2 |
+
FROM python:3.10-slim
|
|
|
3 |
|
4 |
+
#FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
|
|
|
|
5 |
|
6 |
# Install Python and pip
|
7 |
+
#RUN apt-get update && apt-get install -y python3-pip python3-venv
|
8 |
|
9 |
# Where we'll copy the code
|
10 |
WORKDIR /code
|
|
|
13 |
COPY ./requirements.txt /code/requirements.txt
|
14 |
|
15 |
# Install pip requirements without venv
|
16 |
+
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
17 |
|
18 |
# Create a virtual environment and install pip requirements
|
19 |
+
# RUN python3 -m venv /code/venv
|
20 |
+
# RUN /code/venv/bin/pip install --no-cache-dir --upgrade pip
|
21 |
+
# RUN /code/venv/bin/pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
22 |
+
|
23 |
+
# Set the PATH to include the virtual environment's bin directory
|
24 |
+
# ENV PATH="/code/venv/bin:$PATH"
|
25 |
|
26 |
# Creates a non-root user with an explicit UID
|
27 |
# For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers
|
requirements.txt
CHANGED
@@ -1,9 +1,8 @@
|
|
1 |
-
|
|
|
|
|
2 |
transformers
|
3 |
-
torch==2.0.1+cu118
|
4 |
gradio
|
5 |
-
#spaces
|
6 |
huggingface_hub
|
7 |
optimum
|
8 |
-
#for running auto-gptq
|
9 |
auto-gptq
|
|
|
1 |
+
#--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
+
#torch==2.0.1+cu118
|
3 |
+
torch
|
4 |
transformers
|
|
|
5 |
gradio
|
|
|
6 |
huggingface_hub
|
7 |
optimum
|
|
|
8 |
auto-gptq
|