Tensorflow and CUDA version

[Neronjust2017]

Hi, what Tensorflow and CUDA version do you use? When I run the code with the tfrecords data provided in https://github.com/search?q=SIFA, Tensorflow 1.10.0 and CUDA 9.0, the following error occurs. Do you know why? Thanks a lot!

('In the epoch ', 0)
Processing batch 0/2400
2021-11-15 15:49:33.845747: E tensorflow/stream_executor/cuda/cuda_blas.cc:647] failed to run cuBLAS routine cublasSgemmBatched: CUBLAS_STATUS_EXECUTION_FAILED
2021-11-15 15:49:33.845811: E tensorflow/stream_executor/cuda/cuda_blas.cc:2510] Internal: failed BLAS call, see log for details
2021-11-15 15:49:33.848679: I tensorflow/stream_executor/stream.cc:1960] stream 0x5c4c3cf0 did not wait for stream: 0x5c4c3a70
2021-11-15 15:49:33.848703: I tensorflow/stream_executor/stream.cc:4793] stream 0x5c4c3cf0 did not memcpy host-to-device; source: 0x7f4430646800
2021-11-15 15:49:33.848752: F tensorflow/core/common_runtime/gpu/gpu_util.cc:339] CPU->GPU Memcpy failed
2021-11-15 15:49:33.848829: I tensorflow/stream_executor/stream.cc:1960] stream 0x5c4c3cf0 did not wait for stream: 0x5c4c3a70
2021-11-15 15:49:33.848897: I tensorflow/stream_executor/stream.cc:4793] stream 0x5c4c3cf0 did not memcpy host-to-device; source: 0x7f4430706800