error when using TDC dataset to predict DTI #6

New issue

Open

opened 2025-10-14 17:38:00 -06:00 by navan · 0 comments

navan commented

2025-10-14 17:38:00 -06:00

Owner

Originally created by @XWcode11 on 3/30/2024

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

*Originally created by @XWcode11 on 3/30/2024* 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 "<ipython-input-2-0828782a1b59>", line 1, in <module> 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

No labels

bug

enhancement

No milestone

No project

No assignees

1 participant

Notifications

Due date

The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference: github/DeepPurpose#6

No description provided.

Rows
Columns