In order to build your own Dockerfile.
For example to use tensorflow module:
- The following file will be used as an entry point script:
| Code Block | ||||||||
|---|---|---|---|---|---|---|---|---|
| ||||||||
# Module import
import tensorflow as tf
# Use of module
def compute_mean(data):
func = tf.reduce_mean(data)
print(tf.Session().run(func))
# Script
if __name__ == '__main__':
compute_mean(tf.constant([1, 5, 9, 6, 7]))
|
...