Entrypoint bash -l; WorkingDir /home/cudaq

assets=assets vscode_config=docker/release/config/.vscode /bin/bash -c chown -R cudaq /home/cudaq &&
  chgrp -R cudaq /home/cudaq

assets=assets vscode_config=docker/release/config/.vscode /bin/bash -c echo -e '#! /bin/bash \n    jupyter-lab --no-browser --ip=* --ServerApp.allow_origin=* --IdentityProvider.token="$@" \n    ' > "$CUDA_QUANTUM_PATH/bin/jupyter-lab-setup" &&
  chmod +x "$CUDA_QUANTUM_PATH/bin/jupyter-lab-setup"

assets=assets vscode_config=docker/release/config/.vscode /bin/bash -c echo -e '#! /bin/bash \n    if [ ! -x "$(command -v code)" ]; then \n        os=$([ "$(uname -m)" == "aarch64" ] && echo cli-alpine-arm64 || echo cli-alpine-x64) \n        curl -Lk "https://code.visualstudio.com/sha/download?build=stable&os=$os" --output vscode_cli.tar.gz \n        tar -xf vscode_cli.tar.gz && rm vscode_cli.tar.gz && sudo mv code /usr/bin/ \n    fi \n    code "$@"' > "$CUDA_QUANTUM_PATH/bin/vscode-setup" &&
  chmod +x "$CUDA_QUANTUM_PATH/bin/vscode-setup"

docker/release/config/.vscode /home/cudaq/.vscode

vscode_config=docker/release/config/.vscode

assets=assets /bin/bash -c cuda_major_version=$(echo ${CUDA_VERSION} | cut -d . -f1) &&
  if [ "$cuda_major_version" = "12" ]; then pip install --no-cache-dir nvidia-cuda-runtime-cu12 nvidia-cuda-nvrtc-cu12 nvidia-curand-cu12; elif [ "$cuda_major_version" = "13" ]; then pip install --no-cache-dir nvidia-cuda-runtime nvidia-cuda-nvrtc nvidia-curand; else echo "Unsupported CUDA major version: ${cuda_major_version}" &&
  exit 1; fi

assets=assets /bin/bash -c if [ -x "$(command -v pip)" ]; then apt-get install -y --no-install-recommends gcc libpython3-dev &&
  pip install --no-cache-dir -r /tmp/requirements.txt; if [ -n "$MPI_ROOT" ]; then pip install --no-cache-dir mpi4py~=4.1; fi; fi &&
  rm /tmp/requirements.txt

./requirements.txt /tmp/requirements.txt

assets=assets /bin/bash -c cuda_version_suffix=$(echo ${CUDA_VERSION} | tr . -) &&
  for cudart_dependency in libcusolver libcusparse libcublas libcurand cuda-cudart cuda-nvrtc; do if [ -z "$(apt list --installed | grep -o ${cudart_dependency}-${cuda_version_suffix})" ]; then apt-get install -y --no-install-recommends ${cudart_dependency}-${cuda_version_suffix}; fi done &&
  if [ $(echo ${CUDA_VERSION} | cut -d . -f1) -gt 11 ]; then apt-get install -y --no-install-recommends libnvjitlink-${cuda_version_suffix}; fi &&
  apt-get install -y --no-install-recommends curl jq