install.sh 371 Bytes
Newer Older
moto's avatar
moto committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/usr/bin/env bash

unset PYTORCH_VERSION
# For unittest, nightly PyTorch is used, and we do not want to fixiate on the version

set -e

eval "$(./conda/bin/conda shell.bash hook)"
conda activate ./env

printf "* Installing PyTorch nightly build"
conda install -c pytorch-nightly pytorch cpuonly

printf "* Setting up torchaudio\n"
IS_CONDA=true python setup.py develop