devbio-napari icon indicating copy to clipboard operation
devbio-napari copied to clipboard

Published 20 hours ago • verified publisher icon haesleinhuepf

torch dependecy

Open jo-mueller opened this issue 3 years ago • 2 comments
trafficstars

Hi @haesleinhuepf ,

One minor comment/question about the dependencies: When I install devbio-napari, pip automatically downloads torch as well. I think this would be one of the packages that is also best installed through conda since it requires a few more packages to work with GPUS:

According to the documentation, this would look like this:

conda install pytorch torchvision cudatoolkit=11.3 -c pytorch

torch-functions probably run without these packages, but not with GPU support, I believe.

jo-mueller avatar Feb 22 '22 13:02 jo-mueller

Hi @jo-mueller ,

do you know what happens if one installs cudatoolkit=11.3 on a computer that has no NVidia GPU?

haesleinhuepf avatar Feb 23 '22 22:02 haesleinhuepf

Hey @haesleinhuepf ,

do you know what happens if one installs cudatoolkit=11.3 on a computer that has no NVidia GPU?

haven't tried it, but I would suspect that it throws an error during the installation. torch will also not necessarily complain about not having a GPU at its disposal (provided its used in a way that checks for the presence of a GPU first). It could be added as an optional feature in the Readme somewhere further down.

jo-mueller avatar Feb 28 '22 10:02 jo-mueller