lucasnewman / best-rq-pytorch

Implementation of BEST-RQ - a model for self-supervised learning of speech signals using a random projection quantizer, in Pytorch.
MIT License
71 stars 7 forks source link

The vision of CUDA torch torchaudio torchvision #4

Open a897456 opened 9 months ago

a897456 commented 9 months ago

Can you tell me the vision of 'CUDA' 'torch' 'torchaudio' 'torchvision'. I find "orch.cuda.is_available()" is False. Then i change the vision of 'torch' 'torchaudio' and 'torchvision' according to the version of 'CUDA'. Then "orch.cuda.is_available()" is True. Then appear "ImportError" about "torchaudio"