45 lines
1.6 KiB
Bash
Executable File
45 lines
1.6 KiB
Bash
Executable File
#!/bin/bash
|
|
#############################################################
|
|
# use tensorflow to show availabel GPU devices #
|
|
# optional argument is the conda environment to use #
|
|
# default is tf-gpu #
|
|
#############################################################
|
|
TF_ENV=${1:-tf-gpu}
|
|
if [ ${TF_ENV} = "-h" ] ; then
|
|
echo "Usage: $(basename $0) [tensor_flow_env]"
|
|
exit 0
|
|
fi
|
|
|
|
#############################################################
|
|
# set qsub options #
|
|
#############################################################
|
|
#$ -cwd
|
|
#$ -N CUDAtest
|
|
#$ -l memory=64G,h_vmem=64G
|
|
|
|
#############################################################
|
|
# initialize conda #
|
|
#############################################################
|
|
__conda_setup="$('/opt/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
|
|
if [ $? -eq 0 ]; then
|
|
eval "$__conda_setup"
|
|
else
|
|
if [ -f "/opt/anaconda3/etc/profile.d/conda.sh" ]; then
|
|
. "/opt/anaconda3/etc/profile.d/conda.sh"
|
|
else
|
|
export PATH="/opt/anaconda3/bin:$PATH"
|
|
fi
|
|
fi
|
|
unset __conda_setup
|
|
# <<< conda initialize <<<
|
|
|
|
#############################################################
|
|
# activate conda env ent call python commands #
|
|
#############################################################
|
|
conda activate ${TF_ENV}
|
|
export TF_CPP_MIN_LOG_LEVEL=3
|
|
export TF_FORCE_GPU_ALLOW_GROWTH=true
|
|
echo "CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-''}."
|
|
python3 -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())"
|
|
conda deactivate
|