Install PyTorch on CI
Install PyTorch in Travis Python 3.8 CI build
@@ -71,6 +71,11 @@ if [[ $PYTHON == 3.6 ]]; then
conda install -c conda-forge -c defaults contextvars
fi
+if [[ $PYTHON == 3.8 ]]; then
+ # Install pytorch to run related tests
+ conda install -c pytorch -c conda-forge -c defaults pytorch torchvision
+fi
+
if [[ $PYTHON != 3.8 ]]; then
# Install backport package for pickle protocol 5 support
conda install -c conda-forge -c defaults 'pickle5>=0.0.11'
