The user manual literally has a setup for Ubuntu, using CUDA & cupy.