Fix cuda...again (#466)

levkk · web-flow · commit fc1501cc935b · 2022-10-27T15:25:28.000-07:00
diff --git a/pgml-extension/docker/install_cuda.sh b/pgml-extension/docker/install_cuda.sh
@@ -10,12 +10,12 @@ else
 fi
 
 # ARM
-if [[ "$(uname)" == "aarch64" ]]; then
+if [[ ${ARCH} == "aarch64" ]]; then
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/sbsa/cuda-keyring_1.0-1_all.deb
 fi
 
 # Intel
-if [[ "$(uname)" != "aarch64" ]]; then
+if [[ ${ARCH} == "amd64" ]]; then
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
 fi
 
