Open lxy1998 opened 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!
When I run the code below ,it throws an error CUBLAS_STATUS_INVALID_VALUE and terminates. tsnecuda.test() can run Correctly.
Error below:
Please tell me what causes this error and how to fix it, thank you!