Skip to content

Commit e1bf2b2

Browse files
authored
Update dependencies installation in dgx_spark.md
1 parent 3f371c3 commit e1bf2b2

1 file changed

Lines changed: 0 additions & 6 deletions

File tree

docs/dgx_spark.md

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,14 +43,8 @@ COPY requirements.txt .
4343
RUN pip install -r requirements.txt
4444
4545
# Install other major Python libraries in separate layers for better caching
46-
RUN pip install --upgrade torch torchvision
4746
RUN pip install "jax[cuda13-local]==0.7.2"
4847
49-
# Install Transformer Engine and its dependency pybind11
50-
RUN pip install pybind11 && \
51-
export NVTE_FRAMEWORK=jax && \
52-
pip install --no-build-isolation 'transformer_engine[jax]==2.6.0'
53-
5448
# --- Application Code Layer ---
5549
# Now, copy your application source code. This layer is rebuilt only when your code changes.
5650
COPY . .

0 commit comments

Comments
 (0)