IamVicky111 commited on
Commit
dd71798
·
verified ·
1 Parent(s): 6cd135d

Create app.Dockerfile

Browse files
Files changed (1) hide show
  1. app.Dockerfile +20 -0
app.Dockerfile ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM huggingface/transformers:latest
2
+
3
+ RUN apt-get update && apt-get install -y --no-install-recommends \
4
+ build-essential \
5
+ libomp5-dev \
6
+ libopenblas-dev \
7
+ python3-dev \
8
+ wget \
9
+ curl \
10
+ libnvidia-cuda-dev # Adapt based on your CUDA version
11
+ # Add other necessary dependencies here
12
+
13
+ RUN pip install --no-cache-dir torch torchvision torchaudio
14
+
15
+ WORKDIR /app
16
+
17
+ COPY requirements.txt .
18
+ RUN pip install -r requirements.txt # Install additional dependencies
19
+
20
+ COPY . .