提交 2c3bbdcf authored 作者: f0k's avatar f0k

Added note to CUBLAS bug warning in GpuCorrMM docstring

上级 50ba8815
...@@ -791,6 +791,8 @@ class GpuCorrMM(BaseGpuCorrMM): ...@@ -791,6 +791,8 @@ class GpuCorrMM(BaseGpuCorrMM):
shapes. So if you have a Tesla K20, Tesla K40, Quadro K6000, GeForce GT shapes. So if you have a Tesla K20, Tesla K40, Quadro K6000, GeForce GT
640 (DDR5), GeForce GTX 780 (or Ti), GeForce GTX TITAN (or Black or Z) 640 (DDR5), GeForce GTX 780 (or Ti), GeForce GTX TITAN (or Black or Z)
and experience a crash, switching to CUDA 6.5 or CUDA 4.2 should fix it. and experience a crash, switching to CUDA 6.5 or CUDA 4.2 should fix it.
If this is not possible, changing the input or filter shapes (e.g., the
batchsize or number of filters) may also work around the CUBLAS bug.
""" """
def __init__(self, border_mode="valid", def __init__(self, border_mode="valid",
subsample=(1, 1), subsample=(1, 1),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论