With recent Torch nightly builds + TRT GA builds supporting CUDA 12.1, upgrade the stack to support new CUDA version