Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -24,6 +24,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|
| 24 |
python3.8 \
|
| 25 |
python3-pip \
|
| 26 |
python-is-python3 \
|
|
|
|
| 27 |
nvidia-cuda-toolkit \
|
| 28 |
&& rm -rf /var/lib/apt/lists/*
|
| 29 |
|
|
@@ -67,7 +68,7 @@ RUN cd /tmp && pip install -r requirements.txt
|
|
| 67 |
RUN pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu116_pyt1130/download.html
|
| 68 |
RUN python -c "import torch; print(torch.version.cuda)"
|
| 69 |
|
| 70 |
-
RUN chmod 777 $HOME
|
| 71 |
ENV TRANSFORMERS_CACHE=/tmp
|
| 72 |
ENV MPLCONFIGDIR=/tmp
|
| 73 |
|
|
|
|
| 24 |
python3.8 \
|
| 25 |
python3-pip \
|
| 26 |
python-is-python3 \
|
| 27 |
+
python3-opencv \
|
| 28 |
nvidia-cuda-toolkit \
|
| 29 |
&& rm -rf /var/lib/apt/lists/*
|
| 30 |
|
|
|
|
| 68 |
RUN pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu116_pyt1130/download.html
|
| 69 |
RUN python -c "import torch; print(torch.version.cuda)"
|
| 70 |
|
| 71 |
+
RUN chmod -R 777 $HOME
|
| 72 |
ENV TRANSFORMERS_CACHE=/tmp
|
| 73 |
ENV MPLCONFIGDIR=/tmp
|
| 74 |
|