Software » cluster » tensorflow

module load tensorflow/1.15.0
[ -e $CONDA_DIR ] || install_conda
conda create -y -n tensorflow_1.15.0 -c intel intelpython3_full python=3.6
conda activate tensorflow_1.15.0
touch $TENSORFLOW_ENVIRONMENT/easy-install.pth
pip install --upgrade pip setuptools
conda install -y gcc_linux-64 gxx_linux-64 mpi
conda install -y mkl mkl-devel -c intel --no-update-deps
pip install "numpy>=1.14.5,<2.0" "tensorflow-tensorboard<0.2.0,>=0.1.0"
pip install mxnet
pip install tensorflow==1.15.0 opencv-python scikit-image tqdm portpicker keras
python -c "import tensorflow as tf;x1 = tf.constant([1,2,3,4]);x2 = tf.constant([5,6,7,8]);result = tf.multiply(x1, x2);print(result)"
conda deactivate