Last active
September 18, 2021 18:39
-
-
Save Gclabbe/1d85be38b12cfa4f72a40ca2a2199551 to your computer and use it in GitHub Desktop.
4BAI week 6 ... getting it to run locally (Windows)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
## 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