Basically had to install the 570 “open” nvidia drivers for ubuntu to even get my RTX 6000 pro blackwell to work. That package is 12.8 cuda and I have cudnn 9.1 intalled. I didn’t have any luck installing tensorrt. So, I’m trying to build tf 2 and am running into all sorts of issues. Do I just “wait” and have TF team workout the build kinks?
1 Like
Hi @Monty_Hall, A Warm Welcome to the TensorFlow Forum!
Could you please try using tf-nightly build? If that doesn’t fix it, Please wait for the TensorFlow official support. Thank you!
Hi @Divya_Sree_Kayyuri, any update on this? I have tried with both the stable version of TensorFlow and tf-nightly but the gpus (RTX pro 6000 blackwell) are not getting detected. We have upgraded our hardware with RTX pro 6000 blackwell cards and now trying to setup a tensorlfow 2 environment. However, the GPUs are not getting detected. Followed the official instructions given in tensorflow website. Do we have to wait for a supported release from tensorflow? Will building from source help? Is there any update on this? Thank you