ldconfig &&
  apt update &&
  apt upgrade -yqq

dir:5853d94ff71e68f06786914fc2ba3d12517ebe4bc931b6c9ec308b49caf73ec9 in /usr/local/lib/python3.10/dist-packages/

update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1

apt-get update -y &&
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3-numpy &&
  rm -rf /var/lib/apt/lists/*

apt-get update -y &&
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3 &&
  rm -rf /var/lib/apt/lists/*

apt-get update -y &&
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cuda-nvcc-12-0 &&
  rm -rf /var/lib/apt/lists/*

ldconfig

apt-get update &&
  apt-get install -yqq wget &&
  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb &&
  dpkg -i cuda-keyring_1.1-1_all.deb &&
  apt-get update &&
  apt-get install -y cuda-compat-12-0 &&
  ln -s /opt/nvidia/hpc_sdk/Linux_x86_64/23.1/cuda/12.0 /usr/local/cuda

LD_LIBRARY_PATH=/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/12.0/nvshmem/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/nvshmem/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/nccl/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/math_libs/lib64:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/compilers/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/cuda/lib64:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/hpcx/latest/hcoll/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/hpcx/latest/ompi/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/hpcx/latest/nccl_rdma_sharp_plugin/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/hpcx/latest/sharp/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/hpcx/latest/ucx/mt/lib:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/hpcx/latest/ucx/mt/lib/ucx:::/usr/local/openmm/lib/:/opt/nvidia/hpc_sdk/Linux_x86_64/23.1/cuda/12.0/targets/x86_64-linux/lib/

dir:2888aefa7aaafa2b650ea17fd8c1a24015cd63be5afa83e360b94d8aeaca0aea in /opt/nvidia/hpc_sdk/Linux_x86_64/23.1/comm_libs/12.0/nvshmem