facebookresearch / TensorComprehensions

A domain specific language to express machine learning workloads.
https://facebookresearch.github.io/TensorComprehensions/
Apache License 2.0
1.76k stars 211 forks source link

[Build] ATEN Libraries missing when installing TC #597

Closed MaheshRavishankar closed 6 years ago

MaheshRavishankar commented 6 years ago

I just followed the steps listed here https://facebookresearch.github.io/TensorComprehensions/installation.html

This doesnt seem to suggest that ATEN libraries are needed (or how to get them).

So get this error

CMake Error: The following variables are used in this project, but they are set to NOTFOUND. Please set them or make sure they are set and tested correctly in the CMake files: ATEN_LIBRARIES linked by target "tc_autotuner" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/tc/autotuner linked by target "test_basic" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test linked by target "test_mapper_llvm" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test linked by target "test_compile_and_run" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "test_compilation_cache" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "test_autotuner_utility" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "test_tc_mapper_bugs" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "test_corner_cases" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "test_tc_mapper" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "test_autotuner" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/test/cuda linked by target "group_normalization" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/tc/examples linked by target "blockdiagperm" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/tc/examples linked by target "tensordot" in directory /home/mravishankar/TensorComprehensions/TensorComprehensions/tc/examples

skimo-openhub commented 6 years ago

You may need to install an older version of pytorch. Try

conda install pytorch=0.4.0

nicolasvasilache commented 6 years ago

See #598

MaheshRavishankar commented 6 years ago

Thanks for the quick response. pytorch0.4 works.