pycuda
triton==2.1.0
pynvml
