Install torchvision in CUDA CI

Author: zou3519

.circleci/unittest/linux/scripts/install.sh

@@ -56,6 +56,6 @@ if [ "${CU_VERSION:-}" == cpu ] ; then
     conda install -y pytorch torchvision cpuonly -c pytorch-nightly
     PYTORCH_VERSION="$(python -c "import torch; print(torch.__version__)")" python setup.py develop bdist_wheel -d $WHEELS_FOLDER
 else
-    conda install -y pytorch cudatoolkit=10.2 -c pytorch-nightly
+    conda install -y pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly
     PYTORCH_VERSION="$(python -c "import torch; print(torch.__version__)")" python setup.py develop bdist_wheel -d $WHEELS_FOLDER
 fi