pets123 commited on
Commit
0880836
·
verified ·
1 Parent(s): 01115c6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -29,6 +29,7 @@ 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
  # Install system dependencies
34
  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
+ HF_HOME=/app/.cache
33
 
34
  # Install system dependencies
35
  RUN apt-get update && apt-get install -y --no-install-recommends \