yucheng-yu commited on
Commit
09e98cb
·
verified ·
1 Parent(s): 83f30f8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM registry.us-west-1.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda11.8.0-py310-torch2.1.0-tf2.14.0-1.10.0
2
  RUN useradd -m -u 1000 user
3
  USER user
4
  ENV HOME=/home/user \
@@ -13,6 +13,7 @@ RUN echo 'cloning facechain:hf_space_fact'
13
  RUN git clone -b feat/hf_space_fact https://github.com/modelscope/facechain.git
14
  WORKDIR $HOME/facechain
15
  RUN pip install -r requirements.txt
 
16
 
17
  ENV PYTHONPATH=.
18
  RUN python init.py
 
1
+ FROM registry.us-west-1.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-cuda11.7.1-py38-torch2.0.1-tf1.15.5-1.8.1
2
  RUN useradd -m -u 1000 user
3
  USER user
4
  ENV HOME=/home/user \
 
13
  RUN git clone -b feat/hf_space_fact https://github.com/modelscope/facechain.git
14
  WORKDIR $HOME/facechain
15
  RUN pip install -r requirements.txt
16
+ RUN pip install modelscope -U
17
 
18
  ENV PYTHONPATH=.
19
  RUN python init.py