John6666 commited on
Commit
712f3cc
·
verified ·
1 Parent(s): 58e3a70

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -7,7 +7,7 @@ ENV CUDA_HOME="/usr/local/cuda-12.4"
7
  ARG TORCH_CUDA_ARCH_LIST="7.5+PTX"
8
  ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
9
  ENV PATH="/usr/local/cuda/bin:/usr/local/cuda-12/bin:${CUDA_HOME}/bin:${PATH}"
10
- ENV HF_HOME="/data/.huggingface"
11
 
12
  ###########################
13
  # Prepare the environment #
 
7
  ARG TORCH_CUDA_ARCH_LIST="7.5+PTX"
8
  ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
9
  ENV PATH="/usr/local/cuda/bin:/usr/local/cuda-12/bin:${CUDA_HOME}/bin:${PATH}"
10
+ #ENV HF_HOME="/data/.huggingface"
11
 
12
  ###########################
13
  # Prepare the environment #