diff --git a/recipes/TensorFlow-CPU/config/config.yaml b/recipes/TensorFlow-CPU/config/config.yaml index 9dc83755..a06cf1e3 100644 --- a/recipes/TensorFlow-CPU/config/config.yaml +++ b/recipes/TensorFlow-CPU/config/config.yaml @@ -2,4 +2,4 @@ batch_shipyard: storage_account_settings: mystorageaccount global_resources: docker_images: - - tensorflow/tensorflow + - tensorflow/tensorflow:1.15.2 diff --git a/recipes/TensorFlow-CPU/config/jobs.yaml b/recipes/TensorFlow-CPU/config/jobs.yaml index eb972805..a89596ec 100644 --- a/recipes/TensorFlow-CPU/config/jobs.yaml +++ b/recipes/TensorFlow-CPU/config/jobs.yaml @@ -1,8 +1,8 @@ job_specifications: - id: tensorflowjob tasks: - - docker_image: tensorflow/tensorflow + - docker_image: tensorflow/tensorflow:1.15.2 resource_files: - - blob_source: https://raw.githubusercontent.com/tensorflow/models/master/tutorials/image/mnist/convolutional.py + - blob_source: https://raw.githubusercontent.com/tensorflow/models/v1.7.0/tutorials/image/mnist/convolutional.py file_path: convolutional.py command: python -u convolutional.py