为了账号安全,请及时绑定邮箱和手机立即绑定

使用 PyCUDA 后无输出

使用 PyCUDA 后无输出

千万里不及你 2021-09-14 10:30:21
我已经PyCUDA使用pip. 我在两台电脑上试过这个。一个是全新安装的Python 3.7.1,一个是Python 3.6.5.使用后一切都失败了PuCUDA,没有错误消息。最小的例子是这样的:import sysimport pycuda.driver as cudaimport pycuda.autoinit # <-- Comment in order for `print` to workif __name__ == '__main__':    print('Print works')    sys.stdout.write("Sys print works")除非我删除pycuda.autoinit.另一个例子是使用printf:import pycuda.driver as cudaimport pycuda.autoinitfrom pycuda.compiler import SourceModuleif __name__ == '__main__':    mod = SourceModule("""        #include <stdio.h>        __global__ void test() {          printf("I am %d.%d\\n", threadIdx.x, threadIdx.y);        }        """)    func = mod.get_function("test")    func(block=(4, 4, 1))这也不会返回任何输出。我认为 CUDA 失败了,但没有任何报告。我的配置:+--------------------+--------------------+|        PC1         |        PC2         |+--------------------+--------------------+| Python 3.6.5       | Python 3.7.1       || Windows 10         | Windows 10         || Cuda toolkit 9     | Cuda toolkit 10    || GeForce GTX 1050   | GeForce GTX 1080   || Visual Studio 2015 | Visual Studio 2015 |+--------------------+--------------------+驱动程序:GeForce Game Ready DriverVersion        : 418.91 WHQLRelease Date   : Wed Feb 13, 2019我注意到这是一个常见问题,但没有解决方案。
查看完整描述

2 回答

?
红糖糍粑

TA贡献1815条经验 获得超6个赞

我不确定这是否会对您有所帮助,但通过重新安装cuda(使用anoconda)解决了类似的问题。

如果你需要它,你可以通过conda控制台安装它

conda install -c anaconda cudatoolkit

并检查它是否运行良好,

类型 numba -s

希望这可以帮助


查看完整回答
反对 回复 2021-09-14
  • 2 回答
  • 0 关注
  • 245 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信