Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to use GPU train? #69

Open
xinsuinizhuan opened this issue Mar 10, 2022 · 0 comments
Open

how to use GPU train? #69

xinsuinizhuan opened this issue Mar 10, 2022 · 0 comments

Comments

@xinsuinizhuan
Copy link

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Wed_Jul_22_19:09:35_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.221
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28845127_0

VS2019 i set:
auto make_lstm_network() {
return bc::nn::neuralnetwork(
bc::nn::lstm(bc::device_tag(), 96 * 10, 1536, bc::nn::adam),
bc::nn::lstm(bc::device_tag(), 1536, 768, bc::nn::adam),
bc::nn::lstm(bc::device_tag(), 768, 384, bc::nn::adam),
bc::nn::feedforward(bc::device_tag(), 384, 192),
bc::nn::logistic(bc::device_tag(), 192),
bc::nn::logging_output_layer(bc::device_tag(), 192, bc::nn::RMSE).skip_every(100)
);
}
using network_type = decltype(make_lstm_network());
but many compile errors:
图片

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant