Skip to content

Add cuDNN deterministic env variable (only for convolution) - #24301

Closed
duncanriach wants to merge 2 commits into
tensorflow:r1.12from
duncanriach:cudnn_deterministic
Closed

duncanriach wants to merge 2 commits into
tensorflow:r1.12from
duncanriach:cudnn_deterministic

Conversation

@duncanriach

@duncanriach duncanriach commented Dec 11, 2018 •

Copy link
Copy Markdown
Contributor

This change is a major component of the recipe for making TensorFlow training reproducible on GPUs.

Setting the environment variable TF_CUDNN_DETERMINISTIC=1 (or true) will ensure that both forward and backwards convolution algorithms are both fixed and deterministic. It overrides autotune and selects deterministic back-prop algorithms.

@duncanriach
duncanriach changed the base branch from r1.12 to master December 11, 2018 23:45
@duncanriach
duncanriach changed the base branch from master to r1.12 December 11, 2018 23:46
@duncanriach

Copy link
Copy Markdown
Contributor Author

I will to close this pull request for now and open another one with master as base. However, at the time of writing this comment, master is not building for me.

@duncanriach

Copy link
Copy Markdown
Contributor Author

See follow-up pull request based on master.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants