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

yenchenlin / nerf-pytorch

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

MIT License

5.37k stars 1.04k forks source link

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

Open RyanCCC opened 1 year ago

RyanCCC commented 1 year ago

HOW TO SOLVE THIS PROBLEM? Thanks.

97littleleaf11 commented 1 year ago

The pytorch version (cuda 10.2) in your environment is not compatible with the CUDA version 11.6. An easy way to fix this is to create a new conda env and manually install pytorch, for example:

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia