问题描述

跑pytorch相关的代码时，在根据requirements.txt进行环境配置后，运行模型测试代码，出现了以下错误

File "/opt/conda/envs/uof/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__
result = self.forward(*input, **kwargs)
File "/opt/conda/envs/uof/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 87, in forward
return F.linear(input, self.weight, self.bias)
File "/opt/conda/envs/uof/lib/python3.7/site-packages/torch/nn/functional.py", line 1372, in linear
output = input.matmul(weight.t())
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`

在网上查询这个问题的解决方案千奇百怪，尝试之后都没有成功解决。

解决方案

后来猜想应当是pytorch、cuda和python版本不匹配的问题（之前遇到过一次python版本为3.8过高装不上低版本pytorch~=1.3.0的情况），去查找了cuda、pytorch和python之间的版本对应关系表：

https://blog.csdn.net/weixin_45564943/article/details/121688734

根据目前的cuda版本（11.2），python版本（3.7），选择了安装pytorch~=1.7.0版本：

pip uninstall torch==1.3.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

成功解决了该问题

后记

下次再安装pytorch时，一定要考虑python和cuda版本再做选择，不要盲目依赖requirements.txt。

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm 解决方案相关推荐

【bug记录】RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm
问题在训练到一定迭代次数之后报错: RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm ...
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb
这个问题是在测试数据时候遇到的,测试的数据是变长的,没有处理直接送进模型, 有看到说是一次性加载的数据太多(电脑不行),不过也确实一次性加载的数组是(1w+,1,512) 然后修改了一下dataloa ...
【PyTorch】RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm()
完整报错信息. RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa ...
CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm
CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm 在模型测试的时候出现了这个问题, CUDA error: CU ...
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, ...
报错RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasGemmEx( handle, opa,
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasGemmEx( handle, opa, opb ...
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublas‘
调用nn.linear时出现RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublas'错误,搜索网上资料 ...
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) 在写pytorch情感分类 ...
CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, alpha
https://blog.csdn.net/u014176855/article/details/80994991 他的方法对我这个不奏效.检查发现是分类网络的全连接layer第一层神经元的数量设置错 ...

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm 解决方案

目录

问题描述

解决方案

后记

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm 解决方案相关推荐

最新文章

热门文章