RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm解决

在相同的虚拟环境中,一个PyTorch程序在PyCharm中运行正常,但通过命令行启动时出现CUDA错误:CUBLAS_STATUS_INVALID_VALUE。此问题在另一台服务器上并未复现,通过卸载nvidia-cublas-cu11解决了问题。

一,问题描述:

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

        今天遇到的问题很奇怪,在相同的虚拟环境下,运行一个Pytorch的程序,在Pycharm中运行正常,但是通过命令行启动就会报上面的错误。

        而且在另一台服务器上,也是相同的环境,Pytorch版本一致,正常运行,这就导致这个问题更加奇怪。


二,报错详细信息

提示:这里描述项目中遇到的问题:

例如:数据传输过程中数据不时出现丢失的情况,偶尔会丢失一部分数据
APP 中接收数据代码:

Traceback (most recent call last):
  File "/home/dell/anaconda3/envs/CFM/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap
    self.run()
  File "/home/dell/anaconda3/envs/CFM/lib/python3.8/multiprocessing/process.py", line 108, in run
    self._target(*self._args, **sel
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值