AMKCode commited on
Commit
5b1b60b
1 Parent(s): 53acef7

change back to cpu

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
  RUN apt-get update && \
@@ -39,7 +39,7 @@ RUN pyenv install ${PYTHON_VERSION} && \
39
  pyenv global ${PYTHON_VERSION} && \
40
  pyenv rehash && \
41
  pip install --no-cache-dir -U pip setuptools wheel && \
42
- pip install "huggingface-hub" "hf-transfer" "gradio[oauth]==4.28.0" "gradio_huggingfacehub_search==0.0.7" "APScheduler" && \
43
  python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu121 mlc-ai-nightly-cu121
44
 
45
  COPY --chown=1000 . ${HOME}/app
 
1
+ FROM ubuntu:22.04
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
  RUN apt-get update && \
 
39
  pyenv global ${PYTHON_VERSION} && \
40
  pyenv rehash && \
41
  pip install --no-cache-dir -U pip setuptools wheel && \
42
+ pip install "huggingface-hub" "hf-transfer" "gradio[oauth]>=4.28.0" "gradio_huggingfacehub_search==0.0.7" "APScheduler" && \
43
  python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu121 mlc-ai-nightly-cu121
44
 
45
  COPY --chown=1000 . ${HOME}/app