Hugging Face's logo

Spaces:
Yuliang
/
ICON
Runtime error

App Files Community
23
Yuliang commited on
Commit
01cb6e2
·
verified ·
1 Parent(s): 7449886

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -50,7 +50,7 @@ ENV PYTHONUNBUFFERED=1
50
  # ENV FORCE_CUDA=1
51
 
52
  # Set the environment variable to specify the GPU device
53
- ENV CUDA_HOME=/usr/local/cuda:/usr/local:/usr:${CUDA_HOME}
54
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
55
  ENV CUDA_VISIBLE_DEVICES=0
56
 
 
50
  # ENV FORCE_CUDA=1
51
 
52
  # Set the environment variable to specify the GPU device
53
+ ENV CUDA_HOME=/bin/nvcc
54
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
55
  ENV CUDA_VISIBLE_DEVICES=0
56