Open AnouarITI opened 11 months ago
Hi, I figured out that this is a memory issue (That is weird since I am working with the Tesla v100 SXM3). I checked an earlier issue where you suggested modifying lines 53 to 56 in TASO/include/taso/ops.h as follows:
#define MAX_DIM 4
#define MAX_NUM_SPLITS 16
#define MAX_NUM_INPUTS 6
#define MAX_NUM_OUTPUTS 6
But I still get the same issue. Any other suggestion?
Hi, I am new to taso. I am trying to get my first step, so I installed everything as it is supposed to be. Then, I used this example code :
When I try to create a new graph using taso I get this cuda error:
Is this related to the CUDA version installed? This is what I get when I check the nvcc version.
I am using the nvcr.io/nvidia/tensorflow:20.08-tf2-py3 from NVIDIA NGC