Skip to content

Commit

Permalink
Add Python 3.8 binaries for nightlies (pytorch#395)
Browse files Browse the repository at this point in the history
  • Loading branch information
peterjc123 authored and soumith committed Jan 25, 2020
1 parent c637b4b commit cbbe5ed
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 1 deletion.
5 changes: 4 additions & 1 deletion windows/build_pytorch.bat
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,10 @@ FOR %%v IN (%DESIRED_PYTHON%) DO (
set PYTHON_VERSION_STR=%%v
set PYTHON_VERSION_STR=!PYTHON_VERSION_STR:.=!
conda remove -n py!PYTHON_VERSION_STR! --all -y || rmdir %CONDA_HOME%\envs\py!PYTHON_VERSION_STR! /s
conda create -n py!PYTHON_VERSION_STR! -y -q numpy=1.11 "mkl>=2019" cffi pyyaml boto3 cmake ninja typing python=%%v
if "%%v" == "3.5" conda create -n py!PYTHON_VERSION_STR! -y -q numpy=1.11 "mkl>=2019" cffi pyyaml boto3 cmake ninja typing python=%%v
if "%%v" == "3.6" conda create -n py!PYTHON_VERSION_STR! -y -q numpy=1.11 "mkl>=2019" cffi pyyaml boto3 cmake ninja typing python=%%v
if "%%v" == "3.7" conda create -n py!PYTHON_VERSION_STR! -y -q numpy=1.11 "mkl>=2019" cffi pyyaml boto3 cmake ninja typing python=%%v
if "%%v" == "3.8" conda create -n py!PYTHON_VERSION_STR! -y -q numpy=1.11 "mkl>=2019" pyyaml boto3 cmake ninja python=%%v
)
endlocal

Expand Down
9 changes: 9 additions & 0 deletions windows/templates/build_task.yml
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,9 @@ jobs:
PY3.7:
DESIRED_PYTHON: 3.7
CUDA_VERSION: cpu
PY3.8:
DESIRED_PYTHON: 3.8
CUDA_VERSION: cpu
${{ if and(eq(parameters.package, 'Wheels'), eq(parameters.spec, 'CPU')) }}:
LIBTORCH:
DESIRED_PYTHON: 3
Expand All @@ -79,6 +82,9 @@ jobs:
PY3.7_92:
DESIRED_PYTHON: 3.7
CUDA_VERSION: 92
PY3.8_92:
DESIRED_PYTHON: 3.8
CUDA_VERSION: 92
PY3.5_101:
DESIRED_PYTHON: 3.5
CUDA_VERSION: 101
Expand All @@ -88,6 +94,9 @@ jobs:
PY3.7_101:
DESIRED_PYTHON: 3.7
CUDA_VERSION: 101
PY3.8_101:
DESIRED_PYTHON: 3.8
CUDA_VERSION: 101
${{ if and(ne(parameters.spec, 'CPU'), eq(parameters.package, 'Wheels')) }}:
LIBTORCH_92:
DESIRED_PYTHON: 3
Expand Down

0 comments on commit cbbe5ed

Please sign in to comment.