Closed bfreskura closed 7 months ago
Torch actually comes with cuda and cudnn included so we don't need it in the base image.
Reduces image size by ~4GB
https://discuss.pytorch.org/t/does-pytorch-use-the-cudatoolkit-in-docker-or-the-system/140303
Torch actually comes with cuda and cudnn included so we don't need it in the base image.
Reduces image size by ~4GB
https://discuss.pytorch.org/t/does-pytorch-use-the-cudatoolkit-in-docker-or-the-system/140303