enh: add torch CPU install

prepare_release.py

@@ -45,11 +45,11 @@ def get_package_versions(packages, packages_pinned):
     version_dict = {}
     for pkg in packages:
         mod = importlib.import_module(get_import_name(pkg))
-        version_dict[pkg] = mod.__version__
+        version_dict[pkg] = mod.__version__.split("+")[0]
     for pkgv in packages_pinned:
         pk, ver = pkgv.split("==")
         pmod = importlib.import_module(get_import_name(pk))
-        pver = pmod.__version__
+        pver = pmod.__version__.split("+")[0]
         if pver != ver:
             raise ValueError(
                 f"Package {pk} should be {ver}, but is {pver}")

requirements.txt

@@ -1,8 +1,10 @@
 chipstream[CLI]==0.5.0
 dclab==0.59.1
-dcnum==0.22.1
+dcnum==0.23.4
 dcoraid==0.14.2
 h5py==3.11.0
 numpy==1.26.4
 scikit-image==0.23.1  # no-upgrade
-scipy==1.11.4  # no-upgrade
+scipy==1.11.4  # no-upgrade
+torch==2.3.1  # no-upgrade
+torchvision==0.18.1  # no-upgrade

win_build.sh

@@ -17,6 +17,9 @@ python3 -m pip install --upgrade pip wheel
 python3 -m pip install --upgrade pyinstaller
 python3 -m pip install msvc-runtime
 
+# Install CPU versions of torch (smaller file size)
+pip install torch==2.3.1+cpu torchvision-cpu==0.18.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
+
 # upgrade packages
 pip install -r requirements.txt