/test1

Primary LanguageMakefile

I have more Hello world

singularity exec --nv -B /apps2/cuda/10.0/lib64:/home/extraLibs/cuda,/apps2/cudnn/7.1.2/cuda/lib64:/home/extraLibs/cudnn /apps2/singularity/general/general_python3 python3

module load cuda/10.0 cudnn/7.4.2 singularity/3.1
tfSSB=/apps2/singularity/general/general_python3
singularity exec --nv -B /apps2/cuda/10.0/lib64:/home/extraLibs/cuda,/apps2/cudnn/7.4.2/cuda/lib64:/home/extraLibs/cudnn $tfSSB python3 breastCancerClassify2.py MBtrain


//Check GPU
from tensorflow.python.client import device_lib

def get_available_gpus():
    local_device_protos = device_lib.list_local_devices()
    return [x.name for x in local_device_protos if x.device_type == 'GPU']

get_available_gpus()