add torch1.10 to CI

Author: ppwwyyxx

.circleci/config.yml

@@ -226,9 +226,9 @@ workflows:
   regular_test:
     jobs:
       - linux_cpu_tests:
-          name: linux_cpu_tests_pytorch1.9
-          pytorch_version: '1.9.0+cpu'
-          torchvision_version: '0.10.0+cpu'
+          name: linux_cpu_tests_pytorch1.10
+          pytorch_version: '1.10.0+cpu'
+          torchvision_version: '0.11.1+cpu'
       - linux_gpu_tests:
           name: linux_gpu_tests_pytorch1.8
           pytorch_version: '1.8.1+cu102'
@@ -238,10 +238,14 @@ workflows:
           pytorch_version: '1.9+cu102'
           torchvision_version: '0.10+cu102'
       - linux_gpu_tests:
-          name: linux_gpu_tests_pytorch1.9_python39
-          pytorch_version: '1.9+cu102'
-          torchvision_version: '0.10+cu102'
+          name: linux_gpu_tests_pytorch1.10
+          pytorch_version: '1.10+cu102'
+          torchvision_version: '0.11.1+cu102'
+      - linux_gpu_tests:
+          name: linux_gpu_tests_pytorch1.10_python39
+          pytorch_version: '1.10+cu102'
+          torchvision_version: '0.11.1+cu102'
           python_version: '3.9.6'
       - windows_cpu_build:
-          pytorch_version: '1.9+cpu'
-          torchvision_version: '0.10+cpu'
+          pytorch_version: '1.10+cpu'
+          torchvision_version: '0.11.1+cpu'

.github/workflows/workflow.yml

@@ -36,12 +36,14 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        torch: [1.8, 1.9]
+        torch: [1.8, 1.9, 1.10]
         include:
           - torch: 1.8
             torchvision: 0.9
           - torch: 1.9
             torchvision: "0.10"
+          - torch: 1.10
+            torchvision: "0.11.1"
     steps:
       - name: Checkout
         uses: actions/checkout@v2

docker/Dockerfile

@@ -20,7 +20,7 @@ RUN wget https://bootstrap.pypa.io/get-pip.py && \
 # install dependencies
 # See https://pytorch.org/ for other options if you use a different version of CUDA
 RUN pip install --user tensorboard cmake   # cmake from apt-get is too old
-RUN pip install --user torch==1.9 torchvision==0.10 -f https://download.pytorch.org/whl/cu111/torch_stable.html
+RUN pip install --user torch==1.10 torchvision==0.11.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
 
 RUN pip install --user 'git+https://github.com/facebookresearch/fvcore'
 # install detectron2

docker/deploy.Dockerfile

@@ -21,7 +21,7 @@ RUN export CXXFLAGS=-D_GLIBCXX_USE_CXX11_ABI=$(python3 -c 'import torch; print(i
 	./configure --prefix=$HOME/.local && make && make install
 
 # install libtorchvision
-RUN git clone --branch v0.10.0 https://github.com/pytorch/vision/
+RUN git clone --branch v0.11.1 https://github.com/pytorch/vision/
 RUN mkdir vision/build && cd vision/build && \
 	cmake .. -DCMAKE_INSTALL_PREFIX=$HOME/.local -DCMAKE_BUILD_TYPE=Release -DWITH_CUDA=on -DTORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST && \
 	make && make install