Make compatible with Blackwell GPUs
I am using an NVidia RTX 5090 and updated the dockerfile to use a more recent version of CUDA and TensorRT that supports the Blackwell GPUs. It generally seems to run (models compile), but more testing is needed.
I am using an NVidia RTX 5090 and updated the dockerfile to use a more recent version of CUDA and TensorRT that supports the Blackwell GPUs. It generally seems to run (models compile), but more testing is needed.