Skip to content

BLD: add pytorch cpuonly to containers

Update the env file to trigger a rebuild of the environments and close: #739 (closed)

I chose to take this chance to add cpuonly as a dependency so that the CPU version of pytorch is installed. This should make the images smaller, since the CUDA drivers won't be installed.

Merge request reports