Danush-R commited on
Commit
aca3be3
1 Parent(s): dddcfa7

Create dockerfile

Browse files
Files changed (1) hide show
  1. dockerfile +33 -0
dockerfile ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Base image with CUDA and cuDNN
2
+ FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04
3
+
4
+ # Set environment variables for CUDA and cuDNN
5
+ ENV CUDA_VERSION 11.2.2
6
+ ENV CUDNN_VERSION 8
7
+
8
+ # Install dependencies
9
+ RUN apt-get update && apt-get install -y --no-install-recommends \
10
+ build-essential \
11
+ curl \
12
+ ca-certificates \
13
+ python3-pip \
14
+ python3-dev \
15
+ && rm -rf /var/lib/apt/lists/*
16
+
17
+ # Install TensorFlow
18
+ RUN pip3 install --upgrade pip
19
+ RUN pip3 install tensorflow==2.8.0
20
+
21
+ # Install Streamlit and other Python dependencies
22
+ COPY requirements.txt .
23
+ RUN pip3 install -r requirements.txt
24
+
25
+ # Copy the application code
26
+ COPY . /app
27
+ WORKDIR /app
28
+
29
+ # Expose the Streamlit port
30
+ EXPOSE 8501
31
+
32
+ # Run the Streamlit application
33
+ CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]