search

tensorflow / lucid

A collection of infrastructure and tools for research in neural network interpretability.
Apache License 2.0
4.64k stars 652 forks source link

Blas GEMM launch failed #292

Open dokluch opened 2 years ago

dokluch commented 2 years ago

I am running Lucid inside WSL2 Ubuntu 20.04, conda installed environment with python 3.7, tf-gpu 1.15, and the latest lucid. Still, I can't run any of the provided examples. Each returns the same issue:

InternalError: 2 root error(s) found.
  (0) Internal: Blas GEMM launch failed : a.shape=(16384, 3), b.shape=(3, 3), m=16384, n=3, k=3
     [[node MatMul (defined at /home/username/miniconda3/envs/lucid/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]
     [[Mean/_29]]
  (1) Internal: Blas GEMM launch failed : a.shape=(16384, 3), b.shape=(3, 3), m=16384, n=3, k=3
     [[node MatMul (defined at /home/username/miniconda3/envs/lucid/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]
0 successful operations.
0 derived errors ignored.

What might be the issue here?

bhupender-kaushal commented 1 year ago

did you resolve it ? Kindly share your solution if possible........