Are you using the binaries with CUDA11.3 or 11.5? I assume you are not seeing any issues running the same code with a nightly binary on the other GPUs or are you stuck to 1.6.0 in the other setups (this would help identifying if the issue could be in the framework itself or the libraries)?