Skip to content

Move away from pytorch/nvidia channels, use conda-forge exclusively

André Anjos requested to merge issue-88-pytorch-upgrade into main

It turns out that we were using an "unsupported" configuration.

The pytorch package from the pytorch channel should only be used with the defaults channel.

The pytorch package form conda-forge is self contained.

I think it is better to stick with conda-forge, so this MR simply removes all dependencies on pytorch or nvidia channels (closes #88 (closed)).

By using the conda-forge channel, we can also upgrade torchvision to 0.18.1 and replace our custom RGB() transform (closes #91 (closed)).

Edited by André Anjos

Merge request reports