Home

Welcome to the Aurora wiki!

optimizer = SGD(cost, params=[W, b], lr=lr)
for i in range(n_epoch):
    cost_now = optimizer.step(feed_dict={X: x_data, y: y_data})
    if i <= 10 or (i <= 100 and i % 10 == 0) or (i <= 1000 and i % 100 == 0):
        fmt_str = 'iter: {0:>5d} cost: {1:>8.5f}'
        print(fmt_str.format(i, cost_now[0]))

executor = ad.Executor([W, b])
W_val, b_val = executor.run(feed_dict={})

add export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH add export PATH=/usr/local/cuda/bin:$PATH

add export PYTHONPATH="${PYTHONPATH}:/path/to/assignment2/python" go to cuda folder and run the make command.

Next, go test folder and run nosetests -v tests/test_gpu_op.py