Spaces:
Runtime error
Runtime error
File size: 614 Bytes
8aad435 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
FROM docker.io/nvidia/cuda:11.4.3-runtime-ubuntu20.04
MAINTAINER Niv Sardi <x@filtra.me>
WORKDIR /app
RUN apt update && \
export DEBIAN_FRONTEND=noninteractive && \
apt install -y libcairo2 libglib2.0-0 libgl1 python3-pip && \
rm -rf /var/cache/apt
COPY ./yolov5/requirements.txt ./requirements.txt
RUN pip install -r ./requirements.txt --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111
RUN pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111
CMD sh -c 'while true; do echo "🐍 waiting" ; sleep 2h; done'
|