Builds with pytorch 1.5

Summary:
Add conda packages for pytorch 1.5. Make wheels be only pytorch 1.5.

Note that pytorch 1.4 has conda packages for cuda 9.2, 10.0 and 10.1, whilst pytorch 1.5 has packages for cuda 9.2, 10.1 and 10.2. We mirror these choices.

Reviewed By: nikhilaravi

Differential Revision: D21157392

fbshipit-source-id: 2f7311e6a83774a6d6c8afb8110b8bd9f37f1454
This commit is contained in:
Jeremy Reizenstein
2020-04-21 12:12:24 -07:00
committed by Facebook GitHub Bot
parent f2b229c1d1
commit a53a2d3731
4 changed files with 147 additions and 76 deletions

View File

@@ -207,22 +207,27 @@ workflows:
- main
{{workflows()}}
- binary_linux_conda_cuda:
name: testrun_conda_cuda_py3.7_cu100
name: testrun_conda_cuda_py37_cu100_pyt14
python_version: "3.7"
pytorch_version: "1.4"
cu_version: "cu100"
- binary_linux_conda_cuda:
name: testrun_conda_cuda_py37_cu102_pyt15
python_version: "3.7"
pytorch_version: "1.5"
cu_version: "cu102"
- binary_macos_wheel:
cu_version: cpu
name: macos_wheel_py3.6_cpu
name: macos_wheel_py36_cpu
python_version: '3.6'
pytorch_version: '1.4'
pytorch_version: '1.5'
- binary_macos_wheel:
cu_version: cpu
name: macos_wheel_py3.7_cpu
name: macos_wheel_py37_cpu
python_version: '3.7'
pytorch_version: '1.4'
pytorch_version: '1.5'
- binary_macos_wheel:
cu_version: cpu
name: macos_wheel_py3.8_cpu
name: macos_wheel_py38_cpu
python_version: '3.8'
pytorch_version: '1.4'
pytorch_version: '1.5'