Closed ztyree42 closed 5 years ago
Thanks for reporting this @ztyree42!
@gfursin can correct me if I'm wrong, but I don't think that an image based on vanilla Ubuntu 18.04 would support CUDA. Only very recently, thanks to contributions from @me2x, we have started supporting TF with CUDA in Docker containers.
@me2x Could you please provide an example?
Yes, that's right. There is no support for CUDA in the default CK Docker images. But we can be added if we have examples ...
to build cuda enabled TF you need to start from the nvidia docker images. I created a couple of examples in the ck-object-detection repo.
the one linked is using the nvidia container with tensorRT enabled, and compile TF from sources.
Thanks, I'll give it a try.
On Fri, Jul 12, 2019, 5:09 AM Emanuele Vitali notifications@github.com wrote:
to build cuda enabled TF you need to start from the nvidia docker images.
— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/ctuning/ck-tensorflow/issues/130?email_source=notifications&email_token=AECDZPK426W4YQICB2ZQ6YLP7BC5FA5CNFSM4IBII5K2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGODZZGIBA#issuecomment-510813188, or mute the thread https://github.com/notifications/unsubscribe-auth/AECDZPLVJFALXPTH4UE4FZDP7BC5FANCNFSM4IBII5KQ .
@ztyree42 Closing this issue due to lack of activity. Feel free to reopen if you still need assistance. Thanks.
Inside "ctuning/ck-ubuntu-18.04" docker container "ck install package:lib-tensorflow-1.11.0-src-cuda-xla" fails with:
err.txt
Trying again gives different errors (due to race conditions?)
err2.txt