A library for Tensorflow 2 / Keras.
Tested on Python 3.7 with CUDA 11.1 and CUDNN 8.0.4.
Python packages:
- Tensorflow 2.6.0
For CUDA 11.1, if unable to load 'libcusolver.so.10' do the following:
sudo ln -s /usr/local/cuda-11.1/targets/x86_64-linux/lib/libcusolver.so.11 /usr/local/cuda-11.1/targets/x86_64-linux/lib/libcusolver.so.10