adding Dockerfile

Dockerfile

@@ -0,0 +1,27 @@
+FROM --platform=linux/amd64 python:3.10
+
+WORKDIR /code
+
+# Install base utilities
+RUN apt-get update \
+    && apt-get install -y build-essential \
+    && apt-get install -y wget \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install miniconda
+ENV CONDA_DIR /opt/conda
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
+    /bin/bash ~/miniconda.sh -b -p /opt/conda
+
+# Put conda in path so we can use conda activate
+ENV PATH=$CONDA_DIR/bin:$PATH
+
+COPY ./environment.yml /code/environment.yml
+# RUN echo $CONDA_PREFIX
+RUN conda env create -f /code/environment.yml
+    #--file /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

environment.yml

@@ -6,6 +6,7 @@ channels:
   - nvidia/label/cuda-11.8.0
   - pytorch
 dependencies:
+  - cudatoolkit-dev
   - _libgcc_mutex=0.1=conda_forge
   - _openmp_mutex=4.5=2_kmp_llvm
   - asttokens=2.4.1=pyhd8ed1ab_0