Add content to build_conda.sh.

rapidsai · Jan 5, 2024 · a4a19a8 · a4a19a8
1 parent 1b51ade
commit a4a19a8
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/ci/build_conda.sh b/ci/build_conda.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+# Copyright (c) 2024, NVIDIA CORPORATION
+
+set -euo pipefail
+
+source rapids-env-update
+
+export CMAKE_GENERATOR=Ninja
+
+rapids-print-env
+
+rapids-logger "Install CUDA Toolkit"
+yum update
+yum install -y epel-release
+yum install -y nvidia-driver-latest-dkms
+yum install -y cuda-toolkit-12-3
+
+rapids-logger "Begin py build"
+
+rapids-conda-retry mambabuild conda/recipes/pynvjitlink
+
+rapids-upload-conda-to-s3 python