Bittensor Register CUDA

May 17, 2022 ยท View on GitHub

Allows for CUDA registration for bittensor using python.

Requirements

Cython
bittensor@feature/cuda_solver  
nvcc (cuda-11.6)
gcc (9.4.0)
sm_61 enabled CUDA GPU (10XX series or higher)

Install

pip install -e ./bittensor_register_cuda

Testing

Testing uses unittest as there is an issue with pytest and Cython

python3 -m unittest test.py

TODO

  • Speed-up transfer to host after finding a solution to POW
  • Perhaps use events/streams

Acknowledgments

https://github.com/rmcgibbo/npcuda-example/
https://github.com/mochimodev/cuda-hashing-algos/