44 lines
1.3 KiB
Docker
44 lines
1.3 KiB
Docker
FROM tensorflow/tensorflow:2.3.0-gpu
|
|
|
|
# why 2.3 ? I looked it up on stack overflow
|
|
# https://stackoverflow.com/questions/50622525/which-tensorflow-and-cuda-version-combinations-are-compatible
|
|
# here is a nice list, which tf version is compatible with which cuda
|
|
# from the cmmand docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
|
|
# you get your installed cuda version running
|
|
|
|
RUN useradd -ms /bin/bash pluritonian
|
|
|
|
COPY Translations.txt /home/pluritonian/Translations.txt
|
|
|
|
COPY test_runwithgen.py /home/pluritonian/test_runwithgen.py
|
|
COPY test_runwithload.py /home/pluritonian/test_runwithload.py
|
|
COPY generateModels.py /home/pluritonian/generateModels.py
|
|
|
|
COPY req.js /home/pluritonian/req.js
|
|
|
|
COPY postcommand /home/pluritonian/postcommand
|
|
|
|
COPY updateDatabase.py /home/pluritonian/updateDatabase.py
|
|
|
|
COPY FASTsearch.py /home/pluritonian/FASTsearch.py
|
|
|
|
COPY fastapi_server.py /home/pluritonian/fastapi_server.py
|
|
|
|
#USER pluritonian
|
|
|
|
WORKDIR /home/pluritonian
|
|
|
|
RUN apt-get update && apt-get install nano
|
|
|
|
RUN pip install joblib scikit-learn hickle==3.4.9 fastapi uvicorn[standard]
|
|
|
|
RUN pip install idna==2.9 python-multipart==0.0.5
|
|
|
|
RUN python generateModels.py
|
|
|
|
# to let the container running:
|
|
|
|
CMD uvicorn --host 0.0.0.0 fastapi_server:app
|
|
|
|
#ENTRYPOINT ["tail"]
|
|
#CMD ["-f","/dev/null"]
|