install tensorflow with GPU support

tf-with-gpu.sh

#conda create an env
conda create -n py376-tf241 ipykernel python=3.7.6
conda activate py376-tf241

# find out current gpu avaliablity

import tensorflow as tf
print(tf.__version__)
print(tf.config.list_pysical_devices())

# cudnn and cudatoolkit
conda list | grep cuddnn
conda list | grep cudatoolkit

# cuda compiler version
nvcc -V

#specific versionof cudnn and cudatoolkit
conda search cudnn

# install the specific version
conda install cudnn=7.6.5=cuda10.1_0

#install gpu compatible tensorflow 
conda install tensorflow-gpu

#check the tf and gpu again
import tensorflow as tf
print(tf.__version__)
print(tf.config.list_physical_devices())

pip install numba