Skip to content

Instantly share code, notes, and snippets.

@pranjalAI

pranjalAI/loading model.py

Created September 11, 2020 08:20
Show Gist options
  • Save pranjalAI/18f338ddadeba46b22ac8a7411c4593d to your computer and use it in GitHub Desktop.
Save pranjalAI/18f338ddadeba46b22ac8a7411c4593d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
loading model.py
!git clone https://github.com/AlexeyAB/darknet
# verify CUDA
!/usr/local/cuda/bin/nvcc --version
# change makefile to have GPU and OPENCV enabled
%cd darknet
!sed -i 's/OPENCV=0/OPENCV=1/' Makefile
!sed -i 's/GPU=0/GPU=1/' Makefile
!sed -i 's/CUDNN=0/CUDNN=1/' Makefile
!sed -i 's/CUDNN_HALF=0/CUDNN_HALF=1/' Makefile
!make
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment