The specific versions of cudnn and cudatoolkit is because they have to match, CUDA performance is actually quite bad reportedly for 9.2 and KataGo, so I selected 10.0, which is the version that the author mentioned he was originally running when he wrote this program. This program also is tested as working on cuda 10.2, but it requires using a version of cudnn that has not been accepted into the nixpkgs repository yet.

@GrahamcOfBorg build katago katagoWithCuda
@GrahamcOfBorg eval "p: (p {}).katago.override { cudaSupport = true; useTcmalloc = true; }"

Please and thank you.

Updated default CUDA Toolkit from 10.0 to 10.1 because that's what the repository readme suggested.

@rasendubi @erikarvstedt @averelld Hi there, you were active on the original pull request for Leela Zero. If it's not too much trouble, can one of you review this pull request?

Great that this is finally being packaged!
I've built an tested several cuda/opencl * tcmalloc combos successfully (builds and runs fine).

I think just defaulting to tcmalloc would be better if we even make it configurable.

Also, the buildbot probably won't evaluate anything with cuda because of it's unfree nature.

Other than the above, LGTM, the quicker we merge the better!

Ugh, I need to be more careful about screwing around with files locally when they still might change later. This should work now.

@GrahamcOfBorg build katago katagoWithCuda

New release, packaged that instead of the old release.

EDIT: Author updated the tag, so the latest force push is updating the sha commit to match the new tag.

This pull request has been mentioned on NixOS Discourse. There might be relevant details there:

https://discourse.nixos.org/t/prs-ready-for-review-may-2019/3032/137

@GrahamcOfBorg build katago katagoWithCuda

Once again into the night we go.

You may have to restrict the platforms: it seems to require SSE, which rules out i686 and aarch64.

Or is there something more specific?

No, I don't think there is anything more specific.
If it's cross-platform try using platforms.x86_64, otherwise restrict it to linux only with "x86_64-linux".

@GrahamcOfBorg build katago katagoWithCuda

I just rebuilt with v1.3.4, with and without cuda still works great.

He released another minor version to fix a few newly discovered bugs. Sorry about the second version jump in this init pull request.

It's a super short build, I like being up to date (and still works for me).

All right, let's merge this. Thank you @OmnipotentEntity!

Thanks for your help @averelld @rnhmjoj <3