search

aws-samples / amazon-eks-machine-learning-with-terraform-and-kubeflow

Distributed training using Kubeflow on Amazon EKS
Apache License 2.0
82 stars 42 forks source link

Update tensorpack container image to support Tensorflow 2.8.0 #36

Closed ajayvohra2005 closed 2 years ago

ajayvohra2005 commented 2 years ago

Need to update tensorpack container/Dockerfile to support Tensorflow 2.8.x with CUDA 11.4, from Tensorflow 1.15.x.

ajayvohra2005 commented 2 years ago

Issue resolved