Upgrade to NumPy 2.0 (#520)

* Upgrade to NumPy 2.0

* bump pytorch

* bump pytorch

* bump pytorch

* bump pytorch

* bump pytorch

* update

* update packages

.github/workflows/basic-tests-old-pytorch.yml

@@ -1,4 +1,4 @@
-name: Test PyTorch 2.0 and 2.6
+name: Test PyTorch 2.2 and 2.6
 
 on:
   push:
@@ -23,7 +23,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        pytorch-version: [ 2.0.1, 2.6.0 ]
+        pytorch-version: [ 2.3.0, 2.6.0 ]
 
     steps:
     - uses: actions/checkout@v4
@@ -35,10 +35,10 @@ jobs:
 
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
+        python -m pip install --upgrade pip setuptools wheel
         pip install pytest nbval
-        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
         pip install torch==${{ matrix.pytorch-version }}
+        pip install -r requirements.txt
         pip install -r ch05/07_gpt_to_llama/tests/test-requirements-extra.txt
 
     - name: Test Selected Python Scripts

requirements.txt

@@ -1,9 +1,9 @@
-torch >= 2.0.1        # all
+torch >= 2.3.0        # all
 jupyterlab >= 4.0     # all
 tiktoken >= 0.5.1     # ch02; ch04; ch05
 matplotlib >= 3.7.1   # ch04; ch05
-tensorflow >= 2.15.0  # ch05
+tensorflow >= 2.18.0  # ch05
 tqdm >= 4.66.1        # ch05; ch07
-numpy >= 1.25, < 2.0  # dependency of several other libraries like torch and pandas
+numpy >= 1.26, < 2.1  # dependency of several other libraries like torch and pandas
 pandas >= 2.2.1       # ch06
 psutil >= 5.9.5       # ch07; already installed automatically as dependency of torch