Skip to content

Commit d1b8a4a

Browse files
czxttklfacebook-github-bot
authored andcommitted
Torchrec gpu stable
Summary: use this Differential Revision: D38185498 fbshipit-source-id: a96f815ad9c1041287cd8823269b976d0b298a73
1 parent 1002d12 commit d1b8a4a

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

.circleci/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ commands:
172172
# "-e" option will activate the development mode (a symlink to the code in ReAgent
173173
# will be created in site-packages directory)
174174
command: |
175-
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
175+
pip install -e .[gym,test,torchrec_gpu] --pre --extra-index-url https://download.pytorch.org/whl/cu113 --progress-bar off
176176
- unless:
177177
condition: << parameters.is_ubuntu_gpu >>
178178
steps:

setup.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ ax = ax-platform
5353

5454
lite = nevergrad>=0.4.3
5555

56-
torchrec_gpu = torchrec-nightly
56+
torchrec_gpu = torchrec
5757

5858
torchrec_cpu = torchrec-nightly-cpu
5959

tox.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ extras =
3939
test
4040
torchrec_gpu
4141
install_command =
42-
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
42+
pip install --pre --extra-index-url https://download.pytorch.org/whl/cu113 {opts} {packages} --progress-bar off
4343

4444

4545
[testenv:circleci_misc_unittest]

0 commit comments

Comments
 (0)