Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -29,7 +29,8 @@ ENV PYTHONFAULTHANDLER=1 \
|
|
29 |
NVIDIA_DRIVER_CAPABILITIES=compute,utility \
|
30 |
TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6" \
|
31 |
TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
|
32 |
-
|
|
|
33 |
|
34 |
# Install system dependencies
|
35 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
|
29 |
NVIDIA_DRIVER_CAPABILITIES=compute,utility \
|
30 |
TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6" \
|
31 |
TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
|
32 |
+
|
33 |
+
ENV HF_HOME=/app/.cache
|
34 |
|
35 |
# Install system dependencies
|
36 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|