onnx-docker icon indicating copy to clipboard operation
onnx-docker copied to clipboard

Providing GPU enabled docker image

Open haixuanTao opened this issue 4 years ago • 1 comments

Providing CUDA and CUDNN driver within docker image simplify compatibility issues.

This is especially true for onnxruntime 1.6 and 1.7 depending respectively on CUDA 10 and 11.

Those images does not try to minimize the image size. I'll be happy to add any simple optimization. But, as ML and DL tend to be heavy in size I believe simplicity is probably more interesting than size.

haixuanTao avatar Apr 22 '21 09:04 haixuanTao

CLA assistant check
All committers have signed the CLA.

CLAassistant avatar Apr 22 '21 09:04 CLAassistant