Cherry-pick 'Use PyTorch RC for unittest (#953)' (#963)

.circleci/config.yml

@@ -391,6 +391,7 @@ jobs:
       - checkout
       - attach_workspace:
           at: third_party
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
 
@@ -433,6 +434,7 @@ jobs:
       - checkout
       - attach_workspace:
           at: third_party
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
 
@@ -470,6 +472,7 @@ jobs:
       name: windows-cpu
     steps:
       - checkout
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
 
@@ -506,6 +509,7 @@ jobs:
       CUDA_VERSION: "10.1"
     steps:
       - checkout
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
 
@@ -544,6 +548,7 @@ jobs:
       - install_build_tools_macos
       - attach_workspace:
           at: third_party
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
 
@@ -582,6 +587,7 @@ jobs:
     resource_class: medium
     steps:
       - checkout
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
 

.circleci/config.yml.in

@@ -391,6 +391,7 @@ jobs:
       - checkout
       - attach_workspace:
           at: third_party
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
           {% raw %}
@@ -433,6 +434,7 @@ jobs:
       - checkout
       - attach_workspace:
           at: third_party
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
           {% raw %}
@@ -470,6 +472,7 @@ jobs:
       name: windows-cpu
     steps:
       - checkout
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
           {% raw %}
@@ -506,6 +509,7 @@ jobs:
       CUDA_VERSION: "10.1"
     steps:
       - checkout
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
           {% raw %}
@@ -544,6 +548,7 @@ jobs:
       - install_build_tools_macos
       - attach_workspace:
           at: third_party
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
           {% raw %}
@@ -582,6 +587,7 @@ jobs:
     resource_class: medium
     steps:
       - checkout
+      - designate_upload_channel
       - generate_cache_key
       - restore_cache:
           {% raw %}

.circleci/unittest/linux/scripts/install.sh

@@ -17,7 +17,7 @@ else
     cudatoolkit="cudatoolkit=${version}"
 fi
 printf "Installing PyTorch with %s\n" "${cudatoolkit}"
-conda install -y -c pytorch-nightly pytorch "${cudatoolkit}"
+conda install -y -c "pytorch-${UPLOAD_CHANNEL}" pytorch "${cudatoolkit}"
 
 printf "* Installing torchaudio\n"
 BUILD_SOX=1 python setup.py install

.circleci/unittest/windows/scripts/install.sh

@@ -17,7 +17,7 @@ else
     cudatoolkit="cudatoolkit=${version}"
 fi
 printf "Installing PyTorch with %s\n" "${cudatoolkit}"
-conda install -y -c pytorch-nightly pytorch "${cudatoolkit}"
+conda install -y -c "pytorch-${UPLOAD_CHANNEL}" pytorch "${cudatoolkit}"
 
 printf "* Installing torchaudio\n"
 python setup.py install