AndreasH96/tensorflow2_9_cuda_quickfix.py

Created June 1, 2022 12:56

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/AndreasH96/6691f412e0dbe1f563b555ff39628e11.js"></script>
Save AndreasH96/6691f412e0dbe1f563b555ff39628e11 to your computer and use it in GitHub Desktop.

Download ZIP

Bypasses the cudart64_110.dll not found error when importing tensorflow with cuda 11.2

Raw

tensorflow2_9_cuda_quickfix.py

	import os
	os.add_dll_directory("C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/bin")
	import tensorflow

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