Hugging Face's logo

Spaces:
NTUST-DDRC
/
gen3c
Build error

App Files Community
elungky commited on
Commit
a752929
·
1 Parent(s): 106eac9

Removed header symlinks before transformer-engine install

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -52,7 +52,7 @@ RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
52
  # Install Transformer Engine separately after PyTorch and cuDNN are in place and headers are linked.
53
  RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
54
  conda activate cosmos-predict1 && \
55
- pip install transformer-engine[pytorch]==1.12.0
56
 
57
  # Make the start.sh script executable.
58
  RUN chmod +x /app/start.sh
 
52
  # Install Transformer Engine separately after PyTorch and cuDNN are in place and headers are linked.
53
  RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
54
  conda activate cosmos-predict1 && \
55
+ #pip install transformer-engine[pytorch]==1.12.0
56
 
57
  # Make the start.sh script executable.
58
  RUN chmod +x /app/start.sh