Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +5 -0
Dockerfile
CHANGED
|
@@ -29,6 +29,11 @@ RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
|
|
| 29 |
# Switch to non-root user
|
| 30 |
USER user
|
| 31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
|
| 33 |
ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
|
| 34 |
ENV FORCE_CUDA=1
|
|
|
|
| 29 |
# Switch to non-root user
|
| 30 |
USER user
|
| 31 |
|
| 32 |
+
# Create virtual environment
|
| 33 |
+
RUN python -m venv /home/user/venv
|
| 34 |
+
ENV VIRTUAL_ENV=/home/user/venv
|
| 35 |
+
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
|
| 36 |
+
|
| 37 |
ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
|
| 38 |
ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
|
| 39 |
ENV FORCE_CUDA=1
|