From dfb42b8053da83480bff4336f3a3828bd6514405 Mon Sep 17 00:00:00 2001
From: Hannes Probst <hannes@deepnote.com>
Date: Wed, 22 Jan 2025 11:07:39 +0100
Subject: [PATCH] Run pip install with constraints.txt

---
 gpu/setup.python.sh | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/gpu/setup.python.sh b/gpu/setup.python.sh
index d1c7dff..4aa3407 100644
--- a/gpu/setup.python.sh
+++ b/gpu/setup.python.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
 
-# 
+#
 # This script allows us to install a python version of choice
-# 
+#
 
 PYTHON_VER=$1
 TF_VERSION=$2
@@ -30,10 +30,7 @@ ln -s /usr/bin/python3 /usr/bin/python
 
 # We need to remove some pre-installed pip packages that end up clashing with our compute-deps
 pip uninstall -y cryptography
-pip install numpy==1.23.4
-
-# install tensorflow
-pip install tensorflow==$TF_VERSION
+pip install numpy==1.23.4 tensorflow==$TF_VERSION -c https://tk.deepnote.com/constraints${PYTHON_VER}.txt
 
 # create the virtual environment in the home directory in the Dockerfile
 # using our upgraded python version