Module not found `sonnet`

[AnaRhisT94]

Hi, after installing sonnet and tensorflow I'm getting the Module not found sonnet error.

I'm using

dm-sonnet        1.36
tensorflow                   2.8.0
tensorflow-estimator         1.15.1
tensorflow-gpu               1.15.5
tensorflow-io-gcs-filesystem 0.24.0
tensorflow-probability       0.8.0

[tomhennigan]

hi @AnaRhisT94 , we haven't had other reports of this so I wonder if this is something specific to your setup. How did you install Sonnet? What OS are you using? Which package manager?

One thing to note from the versions you listed, Sonnet v1.36 was designed for TF 1.X and you have TF 2.8.0 installed. If you want to use TF 2 (for new codebases you almost definitely do want to) then please upgrade Sonnet to >=2.0. Our documentation and installation instructions all cover Sonnet 2 and TF 2.