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/