Change capabilities in docker-compose (#497)

#479
This commit is contained in:
AbdBarho 2023-06-02 21:29:00 +02:00 committed by GitHub
parent 6695c23150
commit 660c098da0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 3 additions and 4 deletions

View file

@ -38,7 +38,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
COPY . /docker/
RUN cp /docker/extra_model_paths.yaml ${ROOT}
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all
ENV NVIDIA_VISIBLE_DEVICES=all
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
EXPOSE 7860
ENTRYPOINT ["/docker/entrypoint.sh"]