Update pytorch to 1.10RC (#288)

Summary: Pull Request resolved: #288

Test Plan: Passing unittests

Reviewed By: d4l3k

Differential Revision: D31814228

Pulled By: aivanou

fbshipit-source-id: b42139ce73f196dee22e846a20b68020064b4557

Author: aivanou

.github/workflows/python-unittests.yaml

@@ -10,11 +10,20 @@ jobs:
   unittest:
     strategy:
       matrix:
-        python-version: [3.7, 3.8, 3.9]
-        platform: [ubuntu-18.04]
         include:
+# TODO: uncomment this when torch.1-10 is released
+#          - python-version: 3.9
+#            platform: macos-latest
+# TODO: remove this when torch.1-10 is released
+          - python-version: 3.7
+            platform: ubuntu-18.04
+            torch-whl: https://download.pytorch.org/whl/test/cu113/torch-1.10.0%2Bcu113-cp37-cp37m-linux_x86_64.whl
+          - python-version: 3.8
+            platform: ubuntu-18.04
+            torch-whl: https://download.pytorch.org/whl/test/cu113/torch-1.10.0%2Bcu113-cp38-cp38-linux_x86_64.whl
           - python-version: 3.9
-            platform: macos-latest
+            platform: ubuntu-18.04
+            torch-whl: https://download.pytorch.org/whl/test/cu113/torch-1.10.0%2Bcu113-cp39-cp39-linux_x86_64.whl
       fail-fast: false
     runs-on: ${{ matrix.platform }}
     steps:
@@ -30,6 +39,8 @@ jobs:
           set -eux
           pip install coverage codecov
           pip install -e .[dev]
+          pip uninstall -y torch
+          pip install ${{ matrix.torch-whl }}
       - name: Run tests
         run: coverage run -m unittest discover --verbose --start-directory . --pattern "*_test.py"
       - name: Coverage

torchx/runtime/container/Dockerfile

@@ -14,6 +14,16 @@ WORKDIR /app
 COPY dev-requirements.txt /app
 RUN pip install -r dev-requirements.txt
 
+# TODO: Remove this and get dependency on torch 1.10 when it is released
+
+RUN pip uninstall -y torch
+
+RUN pip install https://download.pytorch.org/whl/test/cu113/torch-1.10.0%2Bcu113-cp37-cp37m-linux_x86_64.whl
+
+RUN pip uninstall -y torchtext
+
+RUN pip install https://download.pytorch.org/whl/test/torchtext-0.10.0-cp37-cp37m-linux_x86_64.whl
+
 COPY . /app
 
 RUN python setup.py install