MCG-NJU/MixFormer

the cuda version cudatoolkit=10.2

Opened this issue · 0 comments

zhl98 commented

I saw install_ pytorch17.sh, it is pytorch=1.7.0 torchvision=0.8.1 cudatoolkit=10.2. But this cuda version cannot run on A6000. I have changed it to a new version of pytorch and cuda, but pytorch has imported some old version packages in the code. Is there any good solution?