使用Pycuda多线程

2022-04-17 00:00:00 python python-multithreading pycuda

问题描述

我正在尝试使用Pycuda示例MultipleThreads在GPU上运行多个线程。当我运行我的python文件时,我收到以下错误消息:

(/root/anaconda3/) root@109c7b117fd7:~/pycuda# python multiplethreads.py
Exception in thread Thread-5:
Traceback (most recent call last):
  File "/root/anaconda3/lib/python3.6/threading.py", line 916, in _bootstrap_inner
    self.run()
  File "multiplethreads.py", line 22, in run
    test_kernel(self.array_gpu)
  File "multiplethreads.py", line 36, in test_kernel
    """)
TypeError: 'module' object is not callable

-------------------------------------------------------------------
PyCUDA ERROR: The context stack was not empty upon module cleanup.
-------------------------------------------------------------------
A context was still active when the context stack was being
cleaned up. At this point in our execution, CUDA may already
have been deinitialized, so there is no way we can finish
cleanly. The program will be aborted now.
Use Context.pop() to avoid this problem.
-------------------------------------------------------------------
Exception in thread Thread-6:
Traceback (most recent call last):
  File "/root/anaconda3/lib/python3.6/threading.py", line 916, in _bootstrap_inner
    self.run()
  File "multiplethreads.py", line 22, in run
    test_kernel(self.array_gpu)
  File "multiplethreads.py", line 36, in test_kernel
    """)
TypeError: 'module' object is not callable
Aborted
我已尝试将threading的导入方式从import threading更改为from threading import Thread,但错误仍然存在。有谁能看到我遇到的问题是什么吗?


解决方案

问题已解决。只是导入命令中有一个小错误。

而不是:

from pycuda.compiler import SourceModule

我写错了:

import pycuda.compiler as SourceModule

这是一个严重的错误。

相关文章