search

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
692 stars 90 forks source link

ImportError: libtaso_runtime.so #105

Open sandeepnmenon opened 9 months ago

sandeepnmenon commented 9 months ago

I installed the docker image on my cluster with a V100 GPU. From the docker interactive

 nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

But when I run import taso I get

>>> import taso
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/opt/conda/lib/python3.7/site-packages/taso-0.1.0-py3.7-linux-x86_64.egg/taso/__init__.py", line 1, in <module>
    from .core import *
ImportError: libtaso_runtime.so: cannot open shared object file: No such file or directory

I used the tasoml/cuda100 docker image. What could be the issue?