From db82285501bda179d0b627aa383e9c9c9ff7c977 Mon Sep 17 00:00:00 2001 From: Zhengxing Chen Date: Wed, 27 Jul 2022 00:29:50 -0700 Subject: [PATCH] Torchrec gpu stable (#660) Summary: Pull Request resolved: https://github.com/facebookresearch/ReAgent/pull/660 use this Differential Revision: D38185498 fbshipit-source-id: 16dd4698709687a441253630f2bb5db885568c81 --- .circleci/config.yml | 2 +- setup.cfg | 3 ++- tox.ini | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 759335ed..5c33bdeb 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -172,7 +172,7 @@ commands: # "-e" option will activate the development mode (a symlink to the code in ReAgent # will be created in site-packages directory) command: | - pip install -e .[gym,test,torchrec_gpu] --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/nightly/torchrec_nightly/ --progress-bar off + pip install -e .[gym,test,torchrec_gpu] --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/torchrec/ --progress-bar off - unless: condition: << parameters.is_ubuntu_gpu >> steps: diff --git a/setup.cfg b/setup.cfg index 84a1c702..9641d6bd 100644 --- a/setup.cfg +++ b/setup.cfg @@ -53,8 +53,9 @@ ax = ax-platform lite = nevergrad>=0.4.3 -torchrec_gpu = torchrec-nightly +torchrec_gpu = torchrec +# July 2022: torchrec-cpu stable (0.1.1) caused test errors torchrec_cpu = torchrec-nightly-cpu diff --git a/tox.ini b/tox.ini index 52add020..27703e97 100644 --- a/tox.ini +++ b/tox.ini @@ -39,7 +39,7 @@ extras = test torchrec_gpu install_command = - pip install --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/nightly/torchrec_nightly/ {opts} {packages} --progress-bar off + pip install --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/torchrec/ {opts} {packages} --progress-bar off [testenv:circleci_misc_unittest]