Skip to content

Instantly share code, notes, and snippets.

@Gclabbe
Last active September 18, 2021 18:39
Show Gist options
  • Save Gclabbe/1d85be38b12cfa4f72a40ca2a2199551 to your computer and use it in GitHub Desktop.
Save Gclabbe/1d85be38b12cfa4f72a40ca2a2199551 to your computer and use it in GitHub Desktop.
4BAI week 6 ... getting it to run locally (Windows)
## to get Nvidia's Docker container with Jupyter Notebooks and GPU support
> docker run --gpus all -it --rm -v $(realpath path to your work):/tf/notebooks -p 8888:8888 tensorflow/tensorflow:latest-gpu-jupyter
## New requirements-gpu.txt
opencv-python
lxml
tqdm
-e .
## Inside the notebook
!git clone https://github.com/zzh8829/yolov3-tf2
!pip install -r requirements-gpu.txt
os.chdir('yolov3-tf2')
!apt-get update ##[edited]
!apt-get install ffmpeg libsm6 libxext6 -y
!python convert.py (this will fail if you don't execute the -e . from above)
!wget https://pjreddie.com/media/files/yolov3.weights -O data/yolov3.weights
I actually had to run this from the Ubuntu command line, not in the notebook. Did the same for the 2nd wget
After that, the rest of the tutorial should work
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment