diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml index 39897956c..4d48a0afe 100644 --- a/.github/workflows/docs.yml +++ b/.github/workflows/docs.yml @@ -56,6 +56,9 @@ jobs: - name: Install fbgemm run: | conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu + - name: Install torchmetrics + run: | + conda run -n build_binary pip install torchmetrics==1.0.3 - name: Install TorchRec run: | conda run -n build_binary pip install torchrec --index-url https://download.pytorch.org/whl/nightly/cpu