update to torch 1.11, python 3.10

Author: d4l3k

.github/workflows/python-unittests.yaml

@@ -10,7 +10,7 @@ jobs:
   unittest:
     strategy:
       matrix:
-        python-version: [3.7, 3.8, 3.9]
+        python-version: [3.7, 3.8, 3.9, '3.10']
         platform: [ubuntu-18.04]
         include:
           - python-version: 3.9

dev-requirements.txt

@@ -12,12 +12,14 @@ kfp==1.8.9
 moto==3.0.2
 pyre-extensions==0.0.21
 pytest
-pytorch-lightning==1.5.6
-ray[default]==1.9.2
-torch-model-archiver==0.4.2
-torch==1.10.0
-torchserve==0.4.2
-torchtext==0.11.0
-torchvision==0.11.1
+pytorch-lightning==1.5.10
+torch-model-archiver>=0.4.2
+torch>=1.10.0
+torchserve>=0.4.2
+torchtext>=0.11.0
+torchvision>=0.11.1
 ts==0.5.1
 usort==0.6.4
+
+# Ray doesn't support Python 3.10
+ray[default]==1.11.0; python_version < '3.10'

torchx/runtime/container/Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
+FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
 
 WORKDIR /app