Closed jgcaffari1 closed 4 years ago
Yeah that error comes up if you have tensorflow-cpu installed (or I'm guessing a broken tensorflow-gpu install).
Assuming you have a valid CUDA install, you should just be able to use pip to install tensorflow-gpu; i.e.:
pip install tensorflow-gpu==1.X or pip install tensorflow-gpu==2.0.0
since this code works with Tensorflow 2 as well. Note that I do all my work in Linux; I might not be able to help much if you're using Windows and CUDA..
This fixed my problem. Thank you!
On Mon, Nov 11, 2019 at 6:11 PM vlawhern notifications@github.com wrote:
Yeah that error comes up if you have tensorflow-cpu installed (or I'm guessing a broken tensorflow-gpu install).
Assuming you have a valid CUDA install, you should just be able to use pip to install tensorflow-gpu; i.e.:
pip install tensorflow-gpu==1.X or pip install tensorflow-gpu==2.0.0
since this code works with Tensorflow 2 as well. Note that I do all my work in Linux; I might not be able to help much if you're using Windows and CUDA..
— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/vlawhern/arl-eegmodels/issues/14?email_source=notifications&email_token=AFI4D7JSKCQNY2Q3BVWZ7ZLQTHYC3A5CNFSM4JL4G2JKYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEDYSO7Y#issuecomment-552675199, or unsubscribe https://github.com/notifications/unsubscribe-auth/AFI4D7NSMBP74XD2PTYZWDDQTHYC3ANCNFSM4JL4G2JA .
Good to hear! Closing this but feel free to reopen if there are any other issues.
When I try to fit the eegnet model, I get the following error:
InvalidArgumentError: Default AvgPoolingOp only supports NHWC on device type CPU [[{{node average_pooling2d/AvgPool}} = AvgPoolT=DT_FLOAT, _class=["loc:@training/Adam/gradients/average_pooling2d/AvgPool_grad/AvgPoolGrad"], data_format="NCHW", ksize=[1, 1, 1, 4], padding="VALID", strides=[1, 1, 1, 4], _device="/job:localhost/replica:0/task:0/device:CPU:0"]]
I have cuda installed, but it does not appear to be working. What is the appropriate cuda setup for gpu support?