postgresml
diff --git a/‎pgml-extension/Dockerfile
Copy file name to clipboardExpand all lines: pgml-extension/Dockerfile
+1-1Lines changed: 1 addition & 1 deletion b/‎pgml-extension/Dockerfile
Copy file name to clipboardExpand all lines: pgml-extension/Dockerfile
+1-1Lines changed: 1 addition & 1 deletion
diff --git a/‎pgml-extension/docker/install_cuda.sh
Copy file name to clipboardExpand all lines: pgml-extension/docker/install_cuda.sh
+3Lines changed: 3 additions & 0 deletions b/‎pgml-extension/docker/install_cuda.sh
Copy file name to clipboardExpand all lines: pgml-extension/docker/install_cuda.sh
+3Lines changed: 3 additions & 0 deletions
@@ -31,7 +31,7 @@ RUN apt-get update && apt-fast install postgresql-${PGVERSION} libopenblas-dev c
 
 # CUDA
 COPY docker/install_cuda.sh install_cuda.sh
-RUN UBUNTU_VERSION=${UBUNTU_VERSION} bash install_cuda.sh
+RUN bash install_cuda.sh
 
 USER postgres
 
 
@@ -2,6 +2,9 @@
 #
 # Install CUDA.
 #
+set -e
+
+UBUNTU_VERSION=$(lsb_release -a | grep Release | awk '{ print $2 }')
 
 if [[ $(uname -a) == *"aarch64"* ]]; then
     ARCH="arm64"
Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,9 @@`
`2`	`2`	`#`
`3`	`3`	`# Install CUDA.`
`4`	`4`	`#`
	`5`	`+set -e`
	`6`	`+`
	`7`	`+UBUNTU_VERSION=$(lsb_release -a \| grep Release \| awk '{ print $2 }')`
`5`	`8`
`6`	`9`	`if [[ $(uname -a) == "aarch64" ]]; then`
`7`	`10`	`ARCH="arm64"`