我已经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
希望这可以帮助
添加回答
举报
0/150
提交
取消