RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle) 解决方案

背景

报错代码

def project_latents_to_spanned_subspace(latent: torch.FloatTensor, basis: torch.FloatTensor):
    proj_barycentric = (basis @ basis.t()).inverse() @ basis @ latent
    proj_standard = basis.t() @ proj_barycentric

    return proj_barycentric, proj_standard

报错

RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)

进行矩阵操作的时候@报错，在此之前进行了模型训练，并未出现torch、cuda版本不匹配问题，第一时间查看了输入是否正确，排除了输入问题，又查看了其他博客，发现很多都说是torch版本问题，

解决方案

# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

一开始以为是torch版本太高了，就安装了1.8.1版本的，结果还是不行，安装以上版本就可以了，注意选择cuda版本