Skip to content

Commit 6e8cc97

Browse files
committed
remove fbgemm-gpu-nightly instead
1 parent e4c4f17 commit 6e8cc97

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,10 +92,10 @@ def main(argv: List[str]) -> None:
9292
# install_requires.remove("fbgemm-gpu-nightly")
9393
# install_requires.append("fbgemm-gpu-nightly-cpu")
9494
# if "fbgemm-gpu" in install_requires:
95-
install_requires.remove("fbgemm-gpu")
96-
install_requires.append("fbgemm-gpu-cpu==0.5.0rc3")
95+
install_requires.remove("fbgemm-gpu-nightly")
96+
install_requires.append("fbgemm-gpu-cpu==0.5.0rc3")
9797
else:
98-
install_requires.remove("fbgemm-gpu")
98+
install_requires.remove("fbgemm-gpu-nightly")
9999
install_requires.append("fbgemm-gpu==0.5.0rc2")
100100

101101

0 commit comments

Comments
 (0)