Installing requirement and CUDA on a fresh virtual environnement - Githubissues

YuanGongND / ast

Code for the Interspeech 2021 paper "AST: Audio Spectrogram Transformer".

BSD 3-Clause "New" or "Revised" License

1.13k stars 212 forks source link

Installing requirement and CUDA on a fresh virtual environnement #114

Open blackyx35 opened 1 year ago

blackyx35 commented 1 year ago

Hello, I'm trying to use your AST for my research and to test it out, however it seems the requirements are deprecated with pip install

torchvision 0.10.0 is not compatible with torch 1.8.1
cuda installation is not implemented, while ASTmodels tries to use it

so i customed the requirement.txt file, changing torch and torchvision with these lines :

--find-links https://download.pytorch.org/whl/torch_stable.html
torchvision==0.9.1+cu111
--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cu111

Regards, Gildas Cozanet

zuble commented 6 months ago

yep, thats the way for me too

...
torch==1.8.1+cu101
torchvision==0.9.1+cu101
...

pip install -r requirements.txt --find-links https://download.pytorch.org/whl/torch_stable.html