Cannot import tensor_comprehensions in python 3.5 after installed by conda

Hi,
Need help.
conda env: python 3.5.
Installed tensor_comprehensions according to
conda install -c pytorch -c tensorcomp tensor_comprehensions
under the conda env.
Run python (that is python 3.5) under the conda env and
import tensor_comprehensions
Got:
ImportError: No module named 'tensor_comprehensions'

Would someone know how to fix it ? Checked the package is in conda env
lib/python3.6/site-packages/tensor_comprehensions-0.1.1-py3.6.egg-info/
Is the package only for python3.6 ? Why does python3.6 directory appear in python3.5 env ?

Thanks

Solved. https://facebookresearch.github.io/TensorComprehensions/

Buddy, how did you solve it? I ran into similar problem here T T

Follow the instruction from https://facebookresearch.github.io/TensorComprehensions/installation_conda_dep.html
Don’t follow the instruction from pytorch. http://pytorch.org/2018/03/05/tensor-comprehensions.html

1 Like

Thank you~
By the way, I solved my problem by creating a brand new conda-env.