Merge pull request #2877 from mozilla/fix-compute-for-setup.py

Fix .compute for packaged training code
This commit is contained in:
Reuben Morais 2020-04-01 16:35:28 +02:00 committed by GitHub
commit 4d567cef38
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -8,12 +8,10 @@ python3 -m venv /tmp/venv
source /tmp/venv/bin/activate
pip install -U setuptools wheel pip
pip install -r <(grep -v tensorflow requirements.txt)
pip install .
pip uninstall -y tensorflow
pip install tensorflow-gpu==1.14
# Install ds_ctcdecoder package from TaskCluster
pip install $(python3 util/taskcluster.py --decoder)
mkdir -p ../keep/summaries
data="${SHARED_DIR}/data"