File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -35,6 +35,8 @@ REQUIREMENTS="$(grep -v '^ *#\|^torch\|^torchaudio\|^torchvision|^torchtext' $PR
3535echo $REQUIREMENTS > requirements.txt
3636pip install -r requirements.txt
3737pip install pySoundFile
38+ # Force uninstall torch & related packages, we'll install them using conda later.
39+ pip uninstall -y torch torchvision torchtext
3840conda install -yq -c pytorch " cudatoolkit=10.1" pytorch torchvision torchtext
3941conda install torchaudio -c pytorch-test
4042python -m spacy download de
@@ -47,6 +49,7 @@ if [[ "${CIRCLE_JOB}" == *worker_* ]]; then
4749 python $DIR /remove_runnable_code.py intermediate_source/model_parallel_tutorial.py intermediate_source/model_parallel_tutorial.py || true
4850 python $DIR /remove_runnable_code.py advanced_source/static_quantization_tutorial.py advanced_source/static_quantization_tutorial.py || true
4951 python $DIR /remove_runnable_code.py beginner_source/hyperparameter_tuning_tutorial.py beginner_source/hyperparameter_tuning_tutorial.py || true
52+ python $DIR /remove_runnable_code.py beginner_source/audio_preprocessing_tutorial.py beginner_source/audio_preprocessing_tutorial.py || true
5053
5154 export WORKER_ID=$( echo " ${CIRCLE_JOB} " | tr -dc ' 0-9' )
5255 count=0
You can’t perform that action at this time.
0 commit comments