/opt/nvidia/physicsnemo_env.sh

NVIDIA_PRODUCT_NAME=NVIDIA PhysicsNeMo NVIDIA_PHYSICSNEMO_VERSION=26.08 NVIDIA_BUILD_ID=64216065

NVIDIA_PHYSICSNEMO_BUILD_ID=64216065

NVIDIA_PHYSICSNEMO_VERSION=26.08

/bin/bash -c mkdir -p NVIDIA_PhysicsNeMo_accompanying_licenses /opt/nvidia/entrypoint.d &&
  cp -a /tmp/accompanying_licenses/. NVIDIA_PhysicsNeMo_accompanying_licenses/ &&
  cp -a /tmp/13-physicsnemo-versions.sh /opt/nvidia/entrypoint.d/13-physicsnemo-versions.sh &&
  cp -a /tmp/physicsnemo_env.sh /opt/nvidia/physicsnemo_env.sh &&
  chown -R 0:0 NVIDIA_PhysicsNeMo_accompanying_licenses /opt/nvidia/entrypoint.d/13-physicsnemo-versions.sh /opt/nvidia/physicsnemo_env.sh &&
  pip install "Werkzeug>=3.0.3" &&
  pip install "pyarrow>=14.0.1" &&
  FILE="/etc/pip/constraint.txt" &&
  if [ -f "$FILE" ]; then sed -i '/h11/d' "$FILE"; sed -i '/httpcore/d' "$FILE"; sed -i '/setuptools/d' "$FILE"; sed -i '/tornado/d' "$FILE"; sed -i '/jupyter_core/d' "$FILE"; sed -i '/protobuf/d' "$FILE"; sed -i '/pillow/d' "$FILE"; sed -i '/notebook/d' "$FILE"; sed -i '/jupyterlab/d' "$FILE"; sed -i '/mistune/d' "$FILE"; sed -i '/jupyter[_-]server/d' "$FILE"; sed -i '/^uv\b/d' "$FILE"; else echo "File not found: $FILE"; fi &&
  pip install "h11>=0.16.0" "httpcore>=1.0.8" "tornado>=6.5.0" "jupyter_core>=5.8.1" "pillow>=12.1.1" &&
  pip install "protobuf>=6.33.5" "nbconvert>=7.17.0" &&
  pip install "notebook>=7.5.6" "jupyterlab>=4.5.7" "mistune>=3.2.1" "jupyter-server>=2.18.0" &&
  pip install --upgrade "uv>=0.11.14" &&
  pip install "setuptools>=80.10.2" "wheel>=0.46.2"

PHYSICSNEMO_GIT_HASH=a078229

PHYSICSNEMO_GIT_HASH=a078229

TARGETPLATFORM=linux/amd64 PYSPNG_ARM64_WHEEL=pyspng-0.1.2-cp312-cp312-linux_aarch64.whl NUMCODECS_ARM64_WHEEL=unknown ONNXRUNTIME_ARM64_WHEEL=onnxruntime_gpu-1.20.2-cp312-cp312-linux_aarch64.whl TORCH_SCATTER_ARM64_WHEEL=torch_scatter-2.1.2-cp312-cp312-linux_aarch64.whl TORCH_SCATTER_AMD64_WHEEL=torch_scatter-2.1.2-cp312-cp312-linux_x86_64.whl PYGLIB_ARM64_WHEEL=unknown PYGLIB_AMD64_WHEEL=unknown TORCH_CLUSTER_ARM64_WHEEL=torch_cluster-1.6.3-cp312-cp312-linux_aarch64.whl TORCH_CLUSTER_AMD64_WHEEL=torch_cluster-1.6.3-cp312-cp312-linux_x86_64.whl NATTEN_ARM64_WHEEL=natten-0.21.5-cp312-cp312-linux_aarch64.whl NATTEN_AMD64_WHEEL=natten-0.21.5-cp312-cp312-linux_x86_64.whl /bin/bash -c uv pip uninstall mlflow wandb &&
  uv cache clean

TARGETPLATFORM=linux/amd64 PYSPNG_ARM64_WHEEL=pyspng-0.1.2-cp312-cp312-linux_aarch64.whl NUMCODECS_ARM64_WHEEL=unknown ONNXRUNTIME_ARM64_WHEEL=onnxruntime_gpu-1.20.2-cp312-cp312-linux_aarch64.whl TORCH_SCATTER_ARM64_WHEEL=torch_scatter-2.1.2-cp312-cp312-linux_aarch64.whl TORCH_SCATTER_AMD64_WHEEL=torch_scatter-2.1.2-cp312-cp312-linux_x86_64.whl PYGLIB_ARM64_WHEEL=unknown PYGLIB_AMD64_WHEEL=unknown TORCH_CLUSTER_ARM64_WHEEL=torch_cluster-1.6.3-cp312-cp312-linux_aarch64.whl TORCH_CLUSTER_AMD64_WHEEL=torch_cluster-1.6.3-cp312-cp312-linux_x86_64.whl NATTEN_ARM64_WHEEL=natten-0.21.5-cp312-cp312-linux_aarch64.whl NATTEN_AMD64_WHEEL=natten-0.21.5-cp312-cp312-linux_x86_64.whl /bin/bash -c cd /physicsnemo &&
  uv pip install --no-deps .

TARGETPLATFORM=linux/amd64 PYSPNG_ARM64_WHEEL=pyspng-0.1.2-cp312-cp312-linux_aarch64.whl NUMCODECS_ARM64_WHEEL=unknown ONNXRUNTIME_ARM64_WHEEL=onnxruntime_gpu-1.20.2-cp312-cp312-linux_aarch64.whl TORCH_SCATTER_ARM64_WHEEL=torch_scatter-2.1.2-cp312-cp312-linux_aarch64.whl TORCH_SCATTER_AMD64_WHEEL=torch_scatter-2.1.2-cp312-cp312-linux_x86_64.whl PYGLIB_ARM64_WHEEL=unknown PYGLIB_AMD64_WHEEL=unknown TORCH_CLUSTER_ARM64_WHEEL=torch_cluster-1.6.3-cp312-cp312-linux_aarch64.whl TORCH_CLUSTER_AMD64_WHEEL=torch_cluster-1.6.3-cp312-cp312-linux_x86_64.whl NATTEN_ARM64_WHEEL=natten-0.21.5-cp312-cp312-linux_aarch64.whl NATTEN_AMD64_WHEEL=natten-0.21.5-cp312-cp312-linux_x86_64.whl /bin/bash -c cd /tmp &&
  uv pip install --no-cache --no-binary torch-sparse --no-build-isolation "torch-sparse>=0.6.18" &&
  uv pip install -r pyproject.toml --extra cu13 --extra utils-extras --extra mesh-extras --extra datapipes-extras --extra gnns --extra sym &&
  python -c 'import torch, torch_cluster, torch_scatter, torch_sparse; import natten.libnatten; print("Native extension ABI check passed:", torch.__version__)'