error when using TDC dataset to predict DTI

[XWcode11]

when i used TDC （TDC_104_DeepPurpose）to predict DTI，an error raised:

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)。GPU storage is 24GB, A single GPU.
DeBug :
Traceback (most recent call last):
File "/home/user/anaconda3/envs/DeepPurpose/lib/python3.6/site-packages/IPython/core/interactiveshell.py", line 3343, in run_code exec(code_obj, self.user_global_ns, self.user_ns) File "", line 1, in torch._C._nn.linear(input, weight, bias)
input is tensor(242,50), weight is tensor(265,50), bias is None

here is some package version:
cuda 12.0.0 ha804496_1 conda-forge
gpytorch 1.6.0 pypi_0 pypi