跑一个简单测试 pip新版 gpu包 Segmentation fault
Created by: lgone2000
版本为paddlepaddle_gpu-0.14.0.post87-cp27-cp27mu-manylinux1_x86_64.whl 测试程序如下 其中 如果place为cpu,可以正常输出,改成gpu则core。老版本0.13 可以正常跑
def testpaddle014():
place = fluid.CUDAPlace(0)
#place = fluid.CPUPlace()
print 'version', paddle.__version__, place
input = fluid.layers.data(name='input', shape=[3,50,50], dtype='float32')
output = fluid.layers.conv2d(input=input,num_filters=1,filter_size=3,stride=1,padding=1,groups=1,act=None)
#output = fluid.layers.fc(input=input,size=2)
fetch_list = [output.name]
data = np.zeros((2,3,50,50), np.float32)
exe = fluid.Executor(place)
exe.run(fluid.default_startup_program())
outputlist = exe.run(
fluid.default_main_program(),
feed={'input': data},
fetch_list=fetch_list
)
print 'output', outputlist[0].shape