s495728/Detection-of-plant-diseases
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
e9fffa0539
Detection-of-plant-diseases/src
History
Krzysztof Bojakowski 2093f84c5f Small fixes, model wasnt building
2024-05-05 18:36:53 +02:00
..
model Small fixes, model wasnt building 2024-05-05 18:36:53 +02:00
tests Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
wandb_utils Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
__init__.py Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
.python-version Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
Dockerfile Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
gpu_check.py Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
main.py Small fixes, model wasnt building 2024-05-05 18:36:53 +02:00
README.md Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00
requirements.txt Run example with Dockerfile to run the code 2024-05-05 18:23:13 +02:00

README.md

Setup

  1. Install Docker on your local system
  2. Build docker image and run the shell
  3. Get your API key from https://wandb.ai/settings#api, docker will automatically connect to WanDB.
docker build -t gpu api_key="<wandb_api_key>" .
docker run --rm -it --gpus all --entrypoint /bin/bash gpu
  1. To double check if tensorflow is configured properly run python3 gpu_check.py.