CI: Fix pytorch use newer numpy version

d2l-ai · Aug 9, 2023 · ffffd94 · ffffd94
1 parent ad3cf0b
commit ffffd94
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflow_scripts/build_pytorch.sh b/.github/workflow_scripts/build_pytorch.sh
@@ -10,6 +10,7 @@ TARGET_BRANCH="$2" # Eg. 'master' ; if PR raised to master
 CACHE_DIR="$3"  # Eg. 'ci_cache_pr' or 'ci_cache_push'
 
 pip3 install d2l==0.17.6
+pip3 install --upgrade numpy  # numpy version installed by d2l==0.17.6 is too old
 mkdir _build
 
 source $(dirname "$0")/utils.sh