Spaces:
Build error
Build error
Removed --no-cache-dir --no-build-isolation on transformer-engine
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -61,8 +61,8 @@ RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
|
|
| 61 |
# Install Transformer Engine separately after PyTorch and cuDNN are in place and headers are linked.
|
| 62 |
RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
|
| 63 |
conda activate cosmos-predict1 && \
|
| 64 |
-
|
| 65 |
-
pip install --no-cache-dir --no-build-isolation transformer-engine[pytorch]==1.12.0
|
| 66 |
|
| 67 |
# --- Verification Steps ---
|
| 68 |
RUN echo "Verifying Python and Conda installations..."
|
|
|
|
| 61 |
# Install Transformer Engine separately after PyTorch and cuDNN are in place and headers are linked.
|
| 62 |
RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
|
| 63 |
conda activate cosmos-predict1 && \
|
| 64 |
+
pip install transformer-engine[pytorch]==1.12.0
|
| 65 |
+
#pip install --no-cache-dir --no-build-isolation transformer-engine[pytorch]==1.12.0
|
| 66 |
|
| 67 |
# --- Verification Steps ---
|
| 68 |
RUN echo "Verifying Python and Conda installations..."
|