pets123 commited on
Commit
4a7fccb
·
verified ·
1 Parent(s): 0880836

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- HF_HOME=/app/.cache
 
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 \