Hi,
I am attempting to check if tensorflow installed in 3DSlicer is able to access the GPU for its computation. I get the following outputs in “PythonSlicer.exe”
Does anybody know what is missing here? Tensorflow was able to detect the GPU as well as CUDA runtime (cudart64_101.dll). Is it necessary to install cuDNN?
System Info
- Windows 10
- 3D Slicer (4.11.0-2020-09-05 r29359 / 5513baa)
- PythonSlicer: C:\Users<username>\AppData\Local\NA-MIC\Slicer 4.11.0-2020-09-05\bin\PythonSlilcer.exe
- Tensorflow - 2.3.0
- slicer.util.pip_install(‘tensorflow’)
- CUDA Toolkit - 10.1
- GPU - Nvidia GTX 2080Ti (Driver Version = 451.48)