We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f371c3 commit e1bf2b2Copy full SHA for e1bf2b2
1 file changed
docs/dgx_spark.md
@@ -43,14 +43,8 @@ COPY requirements.txt .
43
RUN pip install -r requirements.txt
44
45
# Install other major Python libraries in separate layers for better caching
46
-RUN pip install --upgrade torch torchvision
47
RUN pip install "jax[cuda13-local]==0.7.2"
48
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
-
54
# --- Application Code Layer ---
55
# Now, copy your application source code. This layer is rebuilt only when your code changes.
56
COPY . .
0 commit comments