cuda/nix-python-devenv/hello.py


1
2
3
4
5
6
7
8
9
10
11
12

import numpy as np


def main():
    print("Hello from nix-python-devenv!")
    print("As you can see, C bindings are correct, because numpy works:")
    print("a =", a := np.array([1, 2, 3]))
    print("mean(a) =", a.mean())


if __name__ == "__main__":
    main()