CannyLab / tsne-cuda

GPU Accelerated t-SNE for CUDA with Python bindings
BSD 3-Clause "New" or "Revised" License
1.78k stars 126 forks source link

CUBLAS_STATUS_INVALID_VALUE #120

Open lxy1998 opened 1 year ago

lxy1998 commented 1 year ago

When I run the code below ,it throws an error CUBLAS_STATUS_INVALID_VALUE and terminates. tsnecuda.test() can run Correctly.

from tsnecuda import TSNE
import pickle

if __name__ == '__main__':
    # tsnecuda.test()

    X = pickle.load(open('/home/featurize/data/emb_all_768/emb_all_768.data', 'rb'))
    print(X.shape)
    # (1694135, 768)

    '''t-SNE-cuda'''
    X_tsne = TSNE(n_components=2, perplexity=15, learning_rate=10).fit_transform(X)
    pickle.dump(X_tsne, open("/home/featurize/data/tsne_result.data", 'wb'), -1)

Error below:

CUBLAS error in file '/tsnecuda/src/util/cuda_utils.cu', line 101, error: CUBLAS_STATUS_INVALID_VALUE
terminating!
python: /tsnecuda/src/util/cuda_utils.cu:103: void __CublasSafeCall(cublasStatus_t, const char*, int): Assertion `0' failed.
[2]    23531 abort (core dumped)  python /home/featurize/work/EventTriplesExtraction/t_sne_cuda.py

Please tell me what causes this error and how to fix it, thank you!