I'm creating environments for several different versions of topaz (including 0.3.7) and I've noticed that with a command like:

mamba create -n topaz python=3.11 topaz fsspec pytorch-cuda=11.8 -c tbepler -c pytorch -c nvidia

then some packages like torchvision are installed from conda base or conda-forge and not pytorch. I used the pytorch::package syntax to make sure the pytorch version with the matching CUDA are installed. I've seen it for torchvision and the base pytorch package (even though the correct pytorch-cuda from pytorch was installed).

Something like the command below seem to correct it. I'm not sure if this is my mistake, the environment solver, or the way the dependencies are listed in the topaz meta.yaml, but perhaps this will help others if they run into the issue.

mamba install python=3.11 topaz=0.2.5 'pytorch::pytorch<=2.3.1' 'pytorch::pytorch-cuda=11.8' 'pytorch::torchvision' -c tbepler -c pytorch -c nvidia

The cuda-version metapackage seems to end up at 12.8 but correcting that seems unimportant so far.