提交 f8233cd0 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Update the version check.

上级 5840d042
......@@ -48,10 +48,10 @@ def init_dev(dev, name=None):
if (pygpu.version.major, pygpu.version.minor) < (0, 6):
raise ValueError(
"Your installed version of pygpu is too old, please upgrade to 0.6 or later")
# This is for the C headers API
if pygpu.gpuarray.api_version()[0] < 0:
# This is for the C headers API, we need to match the exact version.
if pygpu.gpuarray.api_version()[0] != 1:
raise ValueError(
"Your installed libgpuarray is too old, please update")
"Your installed libgpuarray is not in sync, please make sure to have the appropriate version")
if dev not in init_dev.devmap:
context = pygpu.init(
dev,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论