Order of installation

[jeanpat]

Hello,
I try to reinstall lightning-flash in clean virtual environment. If I start with:
pip install 'lightning-flash[image]'
The install starts to install the cpu version of pytorch 1.12:

Should I start to install the latest gpu version followed by the install of lightning-flash from source from source?

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
pip install 'git+https://github.com/PyTorchLightning/lightning-flash.git#egg=lightning-flash[image]'

Thanks

[ethanwharris]

Hey @jeanpat yes, I think the right way would be to install pytorch manually first and then install Flash (either from source or pip should both be fine). That should prevent the Flash install from overriding the pytorch version.

Hope that helps 😃

[jeanpat]

Thanks Ethan,
I'll try that, which version of cuda is recommended?

[ethanwharris]

I think all should be fine, but it depends which torch version as I think they just match each torch version to a couple of cuda versions