Skip to content

Commit

Permalink
limit linux installer building to cpu
Browse files Browse the repository at this point in the history
  • Loading branch information
mschwoer committed Sep 10, 2024
1 parent 224bf07 commit b3d8197
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions release/linux/build_installer_linux.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
#!/bin/bash
set -e -u

CPU_OR_GPU=${1:-CPU}

# Build the installer for Linux.
# This script must be run from the root of the repository.

Expand All @@ -10,6 +12,10 @@ rm -rf dist_pyinstaller build_pyinstaller
python -m build
pip install "dist/alphadia-1.7.2-py3-none-any.whl[stable]"

if [ "${CPU_OR_GPU}" != "GPU" ]; then
pip install torch -U --extra-index-url https://download.pytorch.org/whl/cpu
fi

# Creating the stand-alone pyinstaller folder
pyinstaller release/pyinstaller/alphadia.spec --distpath dist_pyinstaller --workpath build_pyinstaller -y

Expand Down

0 comments on commit b3d8197

Please sign in to comment.