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] Trying on ubuntu 18.04 #614

Open tyoc213 opened 5 years ago

tyoc213 commented 5 years ago

Tensor Comprehensions Github Issues Guidelines

When submitting a bug report, please include the following information (where relevant):

In addition, including the following information will also be very helpful for us to diagnose the problem:

$ python
Python 3.6.8 |Anaconda, Inc.| (default, Dec 30 2018, 01:22:34) 
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensor_comprehensions as tc
>>> import torch
>>> lang = """
... def tensordot(float(N, C1, C2, H, W) I0, float(N, C2, C3, H, W) I1) -> (O) {
...     O(n, c1, c3, h, w) +=! I0(n, c1, c2, h, w) * I1(n, c2, c3, h, w)
... }
... """
>>> N, C1, C2, C3, H, W = 32, 512, 8, 2, 28, 28
>>> tensordot = tc.define(lang, name="tensordot")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: define() got an unexpected keyword argument 'name'
>>> I0, I1 = torch.randn(N, C1, C2, H, W).cuda(), torch.randn(N, C2, C3, H, W).cuda()

/home/tyoc213/anaconda3/envs/tc_build/lib/python3.6/site-packages/torch/cuda/__init__.py:114: UserWarning: 
    Found GPU0 GeForce RTX 2080 which requires CUDA_VERSION >= 9000 for
     optimal performance and fast startup time, but your PyTorch was compiled
     with CUDA_VERSION 8000. Please install the correct PyTorch binary
     using instructions from http://pytorch.org

  warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION))
>>> best_options = tensordot.autotune(I0, I1, cache=True)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'tensordot' is not defined
>>> out = tensordot(I0, I1, options=best_options)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'tensordot' is not defined
>>> 

So, I cant install from source correctly.

By the way, what is the state of this project, it seems like there are a lot of pull request and 8 or 9 months of zero work made in this public repo.