22 Commits

All Branches

Search

Author	SHA1	Message	Date
Tyler	98b2b0f4ba	Downgrade to cuda 11.0.3 for libcusolver.so.10	2022-12-07 01:09:25 -05:00
Tyler	bf3131a13d	Use fork for gpu	2022-12-07 00:53:18 -05:00
Tyler	cc9b2b8a09	Fix entrypoint conditionals	2022-12-07 00:51:04 -05:00
Tyler	c93e179aa1	Fix typo	2022-12-07 00:34:48 -05:00
Tyler	cf59c1ffcd	Use tensorflow 2.4.4 and GPU	2022-12-07 00:32:14 -05:00
Tyler	90b40394e8	Use cudnn7 and force 10.1 for cuda	2022-12-06 08:11:09 -05:00
Tyler	d3c455a423	Fix cuda runtime version	2022-12-06 08:02:35 -05:00
Tyler	e22352a592	Install tensorflow-gpu	2022-12-06 07:36:15 -05:00
Tyler	11943a2012	Fix tag for gpu image, parallel building	2022-12-06 06:07:05 -05:00
Tyler	8e8ca068d0	Build GPU-enabled image	2022-12-06 06:05:50 -05:00
Tyler	cc948a1ad6	Change train_incremental to not use h5	2022-12-04 22:22:48 -05:00
Tyler	21e13c00e3	Fix entrypoint	2022-12-04 04:33:06 -05:00
Tyler	b8b5a66865	Add entrypoint	2022-12-04 04:00:34 -05:00
Tyler	50c51f1a35	Add WORKDIR	2022-12-04 03:51:38 -05:00
Tyler	72406bd395	Fix script name	2022-12-04 03:42:12 -05:00
Tyler	1a55e9bcf8	Add conversion script	2022-12-04 03:37:35 -05:00
Tyler	1b7e7aa8b6	Remove requirements, unnecessary	2022-12-04 00:00:20 -05:00
Tyler	7672edefc3	Install requirements first	2022-12-03 23:40:06 -05:00
Tyler	72d8dbcbd0	Just install from pip	2022-12-03 23:16:41 -05:00
Tyler	7e1e390937	Install requirements first?	2022-12-03 22:43:29 -05:00
Tyler	de91d75eb7	Run setup.sh instead	2022-12-03 22:22:25 -05:00
Tyler	608a6fc79f	Initial version	2022-12-03 22:12:39 -05:00