Move install
This commit is contained in:
1 parent
a199adfbd6
commit
d008c7ab0a
1 file changed
+3
-2
+3
-2
@@ -9,9 +9,10 @@ RUN apt-get update && apt-get upgrade -y \
|
|||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
COPY . .
|
COPY . .
|
||||||
ENV CUDA_DOCKER_ARCH=all
|
ENV CUDA_DOCKER_ARCH=all
|
||||||
RUN pip install deepspeed
|
|
||||||
RUN pip install --no-cache-dir -r requirements_nvidia.txt && \
|
RUN pip install --no-cache-dir -r requirements_nvidia.txt && \
|
||||||
pip install --no-cache-dir -r requirements_finetune.txt && \
|
pip install --no-cache-dir -r requirements_finetune.txt && \
|
||||||
pip install --no-cache-dir -r requirements_other.txt
|
pip install --no-cache-dir -r requirements_other.txt && \
|
||||||
|
pip install --no-cache-dir deepspeed
|
||||||
EXPOSE 7851
|
EXPOSE 7851
|
||||||
CMD ["python", "script.py"]
|
CMD ["python", "script.py"]
|
||||||
Reference in new issue
Block a user