Torchrec gpu stable (facebookresearch#660)

Summary: Pull Request resolved: facebookresearch#660 use this Differential Revision: D38185498 fbshipit-source-id: 409aa190e0a0af00d6e99fce2852b8087623216b
czxttkl · Jul 27, 2022 · 87960bf · 87960bf
1 parent 1002d12
commit 87960bf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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]