Adding test on CUDA at the beginning of the example
Closed this issue · 2 comments
ctroupin commented
In the example, it may be relevant to add the test on CUDA
earlier in the notebook or in the script, so that users know if something failed in the installation. Especially if we consider that running under CPU and not GPU will be way slower.
Alexander-Barth commented
Everything before this test can actually does not use CUDA and does not need to be repeated after installing e.g. nvidia kernel driver. Maybe it is better to put this test (also) in the Installation section of README.md?
ctroupin commented
OK I agree it makes sense to keep it where it is in the code.
I've added a couple of lines in the README.md