at least for me it complains my python is too new (arch) so I installed it on a nvidia cuda docker container (which is using 18.04) and for some reason it still

at least for me it complains my python is too new (arch) so I installed it on a nvidia cuda docker container (which is using 18.04) and for some reason it still complains it can't use CUDA but works fine on the CPU although slow