-
Notifications
You must be signed in to change notification settings - Fork 4.3k
Open
Labels
torchrecPR related to the TorchRec libraryPR related to the TorchRec library
Description
the current lines try to install a cuda 12.1 version when the current colab is 12.6 (and so torch remains at 12.6)
changing the index to cu126 fails, because torchrec requires tensordict, which is not in the index?
my proposed solution is to remove the pytorch install line
and add tensordict to torchmetrics line
# !pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U remove
!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu126
!pip3 install torchmetrics==1.0.3 tensordict
!pip3 install torchrec --index-url https://download.pytorch.org/whl/cu126
Metadata
Metadata
Assignees
Labels
torchrecPR related to the TorchRec libraryPR related to the TorchRec library