【error】 argument 1: <class 'TypeError'>: Don't know how to convert parameter 1

最新推荐文章于 2023-10-20 17:30:11 发布

codebrid

最新推荐文章于 2023-10-20 17:30:11 发布

阅读量7k

点赞数

CC 4.0 BY-SA版权

分类专栏：随记

本文链接：https://blog.youkuaiyun.com/ccbrid/article/details/79000037

随记专栏收录该内容

51 篇文章

订阅专栏

本文记录了一次运行PyTorch程序时遇到的CUDA内部错误，并详细展示了错误信息及解决过程。通过调整CUDA_VISIBLE_DEVICES环境变量成功解决了问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天出现了很奇怪的错误：

Traceback (most recent call last):
  File "main_batch.py", line 175, in <module>
    main(my_args)
  File "main_batch.py", line 102, in main
    seq2seq = seq2seq.cuda() 
  File "/。。anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 216, in cuda
    return self._apply(lambda t: t.cuda(device))
  File "。。naconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 146, in _apply
    module._apply(fn)
  File "/。。naconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 146, in _apply
    module._apply(fn)
  File "/。。/anaconda3/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 123, in _apply
    self.flatten_parameters()
  File "。。anaconda3/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 85, in flatten_parameters
    handle = cudnn.get_handle()
  File "。。/anaconda3/lib/python3.6/site-packages/torch/backends/cudnn/__init__.py", line 296, in get_handle
    handle = CuDNNHandle()
  File "。。naconda3/lib/python3.6/site-packages/torch/backends/cudnn/__init__.py", line 110, in __init__
    check_error(lib.cudnnCreate(ctypes.byref(ptr)))
  File "/。。/anaconda3/lib/python3.6/site-packages/torch/backends/cudnn/__init__.py", line 283, in check_error
    raise CuDNNError(status)
torch.backends.cudnn.CuDNNError: 4: b'CUDNN_STATUS_INTERNAL_ERROR'
Exception ignored in: <bound method CuDNNHandle.__del__ of <torch.backends.cudnn.CuDNNHandle object at 0x7fd3cde68898>>
Traceback (most recent call last):
  File "/。。anaconda3/lib/python3.6/site-packages/torch/backends/cudnn/__init__.py", line 114, in __del__
    check_error(lib.cudnnDestroy(self))
ctypes.ArgumentError: argument 1: <class 'TypeError'>: Don't know how to convert parameter 1

如上文所示为报错信息。

百度无果。

由于程序之前还能跑，突然出现了这样的错误有些手足无措。

查看出错行数与信息，发现是cuda内部问题。

于是执行命令

CUDA_VISIBLE_DEVICES=2 python main_batch.py

错误解决。