Hugging Face's logo

Spaces:
Yuliang
/
ICON
Runtime error

App Files Community
23
Yuliang commited on
Commit
04fb059
·
verified ·
1 Parent(s): fb1e4b5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -80,8 +80,8 @@ RUN pip install https://download.is.tue.mpg.de/icon/HF/kaolin-0.11.0-cp38-cp38-l
80
  RUN pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl
81
  RUN pip install pyembree
82
 
83
- # Verify CUDA installation
84
- RUN nvcc --version
85
  RUN pip install git+https://github.com/YuliangXiu/neural_voxelization_layer.git
86
 
87
  ENV TRANSFORMERS_CACHE=/tmp
 
80
  RUN pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl
81
  RUN pip install pyembree
82
 
83
+ CMD ["which", "nvcc"]
84
+ CMD ["nvcc", "--version"]
85
  RUN pip install git+https://github.com/YuliangXiu/neural_voxelization_layer.git
86
 
87
  ENV TRANSFORMERS_CACHE=/tmp