ThomasBlumet commited on
Commit
1dd0e8c
·
1 Parent(s): 4b4390c

return to normal config without cuda

Browse files
Files changed (2) hide show
  1. Dockerfile +10 -10
  2. requirements.txt +3 -4
Dockerfile CHANGED
@@ -1,13 +1,10 @@
1
  # For more information, please refer to https://aka.ms/vscode-docker-python
2
- #FROM python:3.10-slim
3
- FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
4
 
5
- #nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
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
- #RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
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
- --extra-index-url https://download.pytorch.org/whl/cu118
 
 
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