search

Fictionarry / TalkingGaussian

[ECCV'24] TalkingGaussian: Structure-Persistent 3D Talking Head Synthesis via Gaussian Splatting
https://fictionarry.github.io/TalkingGaussian/
261 stars 33 forks source link

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)` #14

Open einsqing opened 4 months ago

einsqing commented 4 months ago

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

wRinnori commented 4 months ago

i have a same problem，have you solved it?

einsqing commented 4 months ago

i have a same problem，have you solved it?

Not solved

wRinnori commented 4 months ago

i have solved it.Just run the command: unset LD_LIBRARY_PATH After that, the problem should be solved. If you have a another problem like this: Not compiled with GPU support it seems that your pytorch3d have no GPU support

wRinnori commented 3 months ago

I am sorry.It seems that not a good solution，because it may cause some other problem

Fictionarry commented 3 months ago

Ensure pytorch and cudatoolkit version in the conda environment match that of your local cuda.

wRinnori commented 3 months ago

Thank you so much! The CUDA version I had previously was 11.3.0. After updating it to 11.3.1, this issue was resolved.

Coooderr commented 3 months ago

Using cuda/11.3 pytorch/1.12.1 python/3.10 (same as ER-NeRF) can work.