提交 1d4f75a7 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix in recent commit in nvcc_compiler

上级 b18629bd
......@@ -139,6 +139,7 @@ class NVCC_compiler(object):
default_to_move_computation_to_gpu=False,
move_shared_float32_to_gpu=False,
enable_cuda=False)
n = theano.sandbox.cuda.use.device_number
p = theano.sandbox.cuda.device_properties(n)
flags.append('-arch=sm_' + str(p['major']) + str(p['minor']))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论