Bump to pytorch 25.06 and newer TE commit (#14423)

* Bump to pytorch 25.06 and newer TE commit

Signed-off-by: Charlie Truong <[email protected]>

* Fix mamba install

Signed-off-by: Charlie Truong <[email protected]>

---------

Signed-off-by: Charlie Truong <[email protected]>

Author: chtruong814

docker/common/install_dep.sh

@@ -244,7 +244,7 @@ mcore() {
 
   build() {
     if [[ "${NVIDIA_PYTORCH_VERSION}" != "" ]]; then
-      pip wheel --no-deps --no-cache-dir --wheel-dir $WHEELS_DIR $MAMBA_DIR
+      pip wheel --no-deps --no-cache-dir --no-build-isolation --wheel-dir $WHEELS_DIR $MAMBA_DIR
       pip wheel --no-deps --no-cache-dir --wheel-dir $WHEELS_DIR $CAUSAL_CONV1D_DIR
     fi
 

requirements/manifest.json

@@ -1,13 +1,13 @@
 {
-  "ngc-pytorch": "nvcr.io/nvidia/pytorch:25.05-py3",
+  "ngc-pytorch": "nvcr.io/nvidia/pytorch:25.06-py3",
   "vcs-dependencies": {
     "apex": {
       "repo": "https://github.com/NVIDIA/Apex",
       "ref": "810ffae374a2b9cb4b5c5e28eaeca7d7998fca0c"
     },
     "transformer_engine": {
       "repo": "https://github.com/NVIDIA/TransformerEngine",
-      "ref": "f05f12c974b37c5bd8dfca3d2d294be53b66abfa"
+      "ref": "b97c2bf7b93be265fd4d60f600718920297049ef"
     },
     "megatron-lm": {
       "repo": "https://github.com/NVIDIA/Megatron-LM",