no module name tensorflow_probability

[devbhanu]

import graph_nets as gp
Traceback (most recent call last):
File "", line 1, in
File "/home/bhanuteja/anaconda3/lib/python3.6/site-packages/graph_nets/init.py", line 21, in
from graph_nets import blocks
File "/home/desktopname/anaconda3/lib/python3.6/site-packages/graph_nets/blocks.py", line 38, in
import sonnet as snt
File "/home/desktopname /anaconda3/lib/python3.6/site-packages/sonnet/init.py", line 42, in
from sonnet.python import custom_getters
File "/home/desktopname/anaconda3/lib/python3.6/site-packages/sonnet/python/custom_getters/init.py", line 21, in
from sonnet.python.custom_getters import bayes_by_backprop
File "/home/desktopname/anaconda3/lib/python3.6/site-packages/sonnet/python/custom_getters/bayes_by_backprop.py", line 94, in
import tensorflow_probability as tfp

[tomas-wood]

Go install it.
https://github.com/tensorflow/probability

[alvarosg]

See this issue.

The current fix is: pip install -U dm-sonnet==1.23

[jhamrick]

For now, we have released a version which pins the version of sonnet to 1.23. Let's keep this issue open though so we remember to unpin it after the problem is fixed upstream.

[IMBurbank]

Hopefully this adds some useful information to the issue.

I haven't checked Colab, but Graph Nets seems to work locally with a fresh install using Tensorflow's Docker image.

I just checked the Docker images I proposed in PR #4 in light of this bug label. Those images only pull from Tensorflow base-images and follow your current install instructions.

FROM tensorflow/tensorflow

RUN pip install \
        graph_nets \
        tensorflow_probability \
...

dm-sonnet isn't explicitly installed, and all of the images run the tests and demos successfully.

For example, using the Graph Nets images I'm currently hosting

docker run --rm -u $(id -u):$(id -g) -p 8888:8888 -it imburbank/graph_nets:latest-demos

I just ran the shortest_path.

[QianSong-Cherry]

ModuleNotFoundError: No module named 'tensorflow.contrib.framework.argsort'

Hi, I want to know which tensorflow version I should install?

[QianSong-Cherry]

ModuleNotFoundError: No module named 'tensorflow.contrib.framework.argsort'

Hi, I want to know which tensorflow version I should install?

I upgraded tensorflow to 1.12, now it works.