Configuring a tensorflow environment to work with gpu's is really annoying. Thankfully, conda & pipmake it a lot less annoying.
Here are the core dependencies for getting this to work.
- cudatoolkit=11.6.0
- cudnn=8.2.1.32
- tensorflow=2.7.0
- tensorflow-gpu=2.7.0
- tensorflow-io=0.25.0
- python=3.9