Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.
Sign upGitHub is where the world builds software
Millions of developers and companies build, ship, and maintain their software on GitHub — the largest and most advanced development platform in the world.
Add GPU support #9
Conversation
distutils cannot compile CUDA code by default. Therefore, this commit uses `make` to build darknet with GPU and CUDNN flags. `GPU=1 pip install lightnet` installs lightnet with GPU support. `CUDNN=1 pip install lightnet` installs lightnet with CUDNN support.
|
Hi @uniq10, How can I compile and install your branch in my local pip? |
|
Hi @kumaranvpl checkout to my branch and do |
|
Hi @uniq10, |
|
@kumaranvpl can you post the full error message? Do you have lightnet already installed? GPU=1 just uses CUDA while CUDNN=1 uses CuDNN too. |
distutils cannot compile CUDA code by default. Therefore, this commit
uses
maketo build darknet with GPU and CUDNN flags.GPU=1 pip install lightnetinstalls lightnet with GPU support.CUDNN=1 pip install lightnetinstalls lightnet with CUDNN support.