I found a solution based off of another link related to the above link you provided.
Adding this to the dockerfile works perfectly and allows the evaluation script to run:
ENV TF_FORCE_GPU_ALLOW_GROWTH=true
Thanks for your help @Morganh.
I found a solution based off of another link related to the above link you provided.
Adding this to the dockerfile works perfectly and allows the evaluation script to run:
ENV TF_FORCE_GPU_ALLOW_GROWTH=true
Thanks for your help @Morganh.