A4000 not supported

[CoWayger]

Hello,
anyone know how to edit Dockerfile to support for newer GPUs? Will this need to also update code somehow?

[aleks648]

You might want to try changing the first line in the docker file from
FROM nvidia/cuda:11.1-base
to
FROM nvidia/cuda:12.1.0-base-ubuntu20.04
Hopefully that will ensure the latest version of pytorch gets installed which should support newer GPUs.