60 Commits

Author SHA1 Message Date
Glenn Jocher
c43439aa31
Add albumentations to Dockerfile (#6392) 2022-01-21 21:06:02 -10:00
Glenn Jocher
bd815d48df
FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6379)
21.12 generates dockerhub errors so rolling back to 21.10 with latest pytorch install. Not sure if this torch install will work on non-GPU dockerhub autobuild so this is an experiment.
2022-01-20 19:15:04 -10:00
Glenn Jocher
1b41a1d059
FROM nvcr.io/nvidia/pytorch:21.12-py3 (#6377) 2022-01-20 17:59:41 -10:00
Glenn Jocher
94d8fec6d8
GCP VM from Image example (#5814) 2021-11-28 18:12:46 +01:00
Glenn Jocher
7207fe95e5
FROM nvcr.io/nvidia/pytorch:21.10-py3 (#5592) 2021-11-09 17:55:57 +01:00
Glenn Jocher
ce8e5dc8bb
Update Dockerfile to ADD Arial.ttf (#5084)
* Update Dockerfile to `ADD` Arial.ttf

Fix for `Arial.ttf download in Docker` #5078

* Cleanup

* Cleanup2
2021-10-07 14:16:33 -07:00
Glenn Jocher
c5ba2abb4a
Update Dockerfile (#4935) 2021-09-26 19:16:14 -07:00
Glenn Jocher
2809616345
Update Dockerfile (#4862) 2021-09-20 13:10:41 +02:00
Glenn Jocher
40d1c80503
Update Dockerfile (#4861) 2021-09-20 12:22:24 +02:00
Ayush Chaurasia
302a1b0bb0
W&B: Enable login timeout (#4843)
* evolve fix

* Enable login timeout

* fix pkg
2021-09-17 18:59:34 +02:00
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings (#4417)
* Implement new headers

* Reformat 1

* Reformat 2

* Reformat 3 - math

* Reformat 4 - yaml
2021-08-14 21:17:51 +02:00
Glenn Jocher
47543f97b1
FROM nvcr.io/nvidia/pytorch:21.05-py3 (#3794) 2021-06-26 17:12:43 +02:00
Glenn Jocher
5e6886c860
Update Dockerfile (#3792) 2021-06-26 16:51:17 +02:00
Glenn Jocher
96c87f1711
nvcr.io/nvidia/pytorch:21.06-py3 (#3791) 2021-06-26 16:28:06 +02:00
Glenn Jocher
07166ba38c
Update Usage examples (#3790) 2021-06-26 16:09:56 +02:00
Glenn Jocher
2754adad46
Remove /weights directory (#3659)
* Remove `/weights` directory

* cleanup
2021-06-17 14:44:12 +02:00
Glenn Jocher
bb79e13d52
RUN pip install --no-cache -U torch torchvision (#3637) 2021-06-16 12:35:33 +02:00
Glenn Jocher
de56813ba8
FROM nvcr.io/nvidia/pytorch:21.05-py3 (#3633) 2021-06-16 10:40:31 +02:00
Glenn Jocher
85b6fdda0f
FROM nvcr.io/nvidia/pytorch:21.03-py3
Issues with 21.04
2021-05-05 01:00:12 +02:00
Glenn Jocher
4c7d686d7e
FROM nvcr.io/nvidia/pytorch:21.04-py3 (#3030) 2021-05-05 00:16:35 +02:00
Glenn Jocher
9ccfa85249
pip install coremltools onnx (#2690)
Requested in https://github.com/ultralytics/yolov5/issues/2686
2021-04-02 13:00:46 +02:00
Glenn Jocher
9b92d3ee76
FROM nvcr.io/nvidia/pytorch:21.03-py3 (#2623)
Update Docker FROM nvcr.io/nvidia/pytorch:21.03-py3
2021-03-27 18:35:53 +01:00
Glenn Jocher
2dfe32030a
Remove conflicting nvidia-tensorboard package (#2622)
Attempt to resolve tensorboard Docker error in https://github.com/ultralytics/yolov5/issues/2573
2021-03-27 18:31:53 +01:00
Glenn Jocher
c64fe219b4
ENV HOME=/usr/src/app (#2382)
Set HOME environment variable per Binder requirements. 
https://github.com/binder-examples/minimal-dockerfile
2021-03-05 15:53:57 -08:00
Glenn Jocher
692e1f31dc
--no-cache notebook (#2381) 2021-03-05 15:26:27 -08:00
Glenn Jocher
ab86cec854
FROM nvcr.io/nvidia/pytorch:21.02-py3 (#2341) 2021-03-02 13:01:59 -08:00
Glenn Jocher
c2026a5f35
Update Dockerfile install htop (#2320) 2021-02-27 15:55:31 -08:00
Glenn Jocher
cc79f3a9ea
Update Dockerfile with apt install zip (#2274) 2021-02-22 22:50:00 -08:00
Glenn Jocher
08d3119e09
Add histogram equalization fcn (#2049) 2021-01-26 17:30:42 -08:00
Glenn Jocher
3551b072b3
Docker pyYAML>=5.3.1 fix (#2031) 2021-01-24 11:54:15 -08:00
Glenn Jocher
8dc68fcbcb
Update Dockerfile (#2016) 2021-01-22 14:26:58 -08:00
Glenn Jocher
1ca2d26b9f
Update Dockerfile (#1982) 2021-01-19 11:45:15 -08:00
Glenn Jocher
b1cf25dd9a
check_git_status() asserts (#1977) 2021-01-18 10:49:28 -08:00
Glenn Jocher
720afe68e9
FROM nvcr.io/nvidia/pytorch:20.12-py3 (#1763) 2020-12-22 20:53:44 -08:00
Glenn Jocher
b2bef8f6d8
vast.ai compatability updates (#1657) 2020-12-10 06:00:47 -08:00
Glenn Jocher
2e8e02745b
vast.ai compatability updates (#1657) 2020-12-09 19:01:08 -08:00
Glenn Jocher
68211f72c9
FROM nvcr.io/nvidia/pytorch:20.10-py3 (#1553)
* FROM pytorch/pytorch:latest

* FROM nvcr.io/nvidia/pytorch:20.10-py3
2020-11-29 17:47:51 +01:00
Glenn Jocher
5dbf0c96f4
FROM nvcr.io/nvidia/pytorch:20.11-py3 2020-11-27 01:28:33 +01:00
Glenn Jocher
4821d076e2
Increment train, test, detect runs/ (#1322)
* Increment train, test, detect runs/

* Update ci-testing.yml

* inference/images to data/images

* move images

* runs/exp to runs/train/exp

* update 'results saved to %s' str
2020-11-08 19:39:05 +01:00
Glenn Jocher
7fe4a6bc06 FROM nvcr.io/nvidia/pytorch:20.10-py3 2020-10-25 10:49:33 +01:00
Glenn Jocher
10c85bf4eb
FROM nvcr.io/nvidia/pytorch:20.09-py3 2020-10-11 16:25:14 +02:00
Glenn Jocher
bb8872ea5f Dockerfile sudo docker "kill all" bug fix 2020-09-08 00:56:35 -07:00
Glenn Jocher
0c01afc390 Update Dockerfile to upgrade pip 2020-08-26 14:44:43 -07:00
Glenn Jocher
62e037419f Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3 2020-08-26 00:44:34 -07:00
Glenn Jocher
2077d78880 update Dockerfile with libgl1-mesa-glx 2020-08-15 09:53:22 -07:00
Glenn Jocher
916d4aad9a
v3.0 Release (#725)
* initial commit

* remove yolov3-spp from test.py study

* update study --img range

* update mAP

* cleanup and speed updates

* update README plot
2020-08-13 14:25:05 -07:00
Glenn Jocher
9f482cbcb8 utils.general comment updates/bug fixes 2020-08-11 20:11:44 -07:00
Glenn Jocher
8bf3cff875
PyTorch 1.6.0 update with native AMP (#573) 2020-07-31 11:10:03 -07:00
Glenn Jocher
0032af2980
Update utils.py strip_optimizer() (#509) 2020-07-24 19:26:56 -07:00
Glenn Jocher
2dd43bc82a update Dockerfile 2020-07-15 17:40:19 -07:00