Hugging Face's logo

Spaces:
qihang
/
BerfScene
Sleeping

App Files Community
3v324v23 commited on
Commit
e73a1b1
·
1 Parent(s): d810ae0

update

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -1,6 +1,8 @@
1
  # FROM nvidia/cuda:11.1.1-devel-ubuntu20.04
2
  FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
3
 
 
 
4
  ENV CUDA_HOME=/usr/local/cuda
5
  ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
6
  ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
1
  # FROM nvidia/cuda:11.1.1-devel-ubuntu20.04
2
  FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
3
 
4
+ ARG USER_NAME=berfscene
5
+
6
  ENV CUDA_HOME=/usr/local/cuda
7
  ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
8
  ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}