The version of Tensorflow-gpu? Some conflicts occur

[ThePassedWind]

Could you show your specific version of tensorflow-gpu? There are some errors by using the default version of it.

Apart from this, there are some conflicts between two package : oryx, jax(jaxlib) if using provided versions in your install.sh

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
oryx 0.2.4 requires jax==0.3.21, but you have jax 0.3.20 which is incompatible.
oryx 0.2.4 requires jaxlib==0.3.20, but you have jaxlib 0.3.15+cuda11.cudnn82 which is incompatible.
oryx 0.2.4 requires tensorflow-probability[jax]==0.18.0, but you have tensorflow-probability 0.20.1 which is incompatible.

Finally, I avoid this conflict by setting them: oryx==0.2.4、jaxlib==0.3.20、jax==0.3.21、tensorflow-probability[jax]==0.18.0