diff --git a/1.7-gpu/README.md b/1.7-gpu/README.md deleted file mode 100644 index 6d02839..0000000 --- a/1.7-gpu/README.md +++ /dev/null @@ -1 +0,0 @@ -This is a trial file diff --git a/1.9-gpu/README b/1.9-gpu/README new file mode 100644 index 0000000..20c213b --- /dev/null +++ b/1.9-gpu/README @@ -0,0 +1,17 @@ +To install tensorflow with virtualenv you have to: + +1) Create the virtualenv environment and choose a Python interpreter + +virtualenv --system-site-packages -p python3 tensorflow-1.9 + +2) Activate the virtualenv environment + +source ~/tensorflow/tensorflow-1.9/bin/activate + +3) Use pip3 to install the package + +pip3 install --no-deps --upgrade --user tensorflow-1.9.0-cp36-cp36m-linux_x86_64.whl + +4) You might need to install extra packages with pip3 + +5) Modify the script.sh provided here and enjoy tensorflow diff --git a/1.9-gpu/script.sh b/1.9-gpu/script.sh new file mode 100644 index 0000000..6b8eb47 --- /dev/null +++ b/1.9-gpu/script.sh @@ -0,0 +1,16 @@ +#!/bin/bash + +#SBATCH --nodes=1 +#SBATCH --time=1:0:0 +#SBATCH --partition=gpu +#SBATCH --gres=gpu:1 +#SBATCH --qos=gpu + +source /ssoft/spack/bin/slmodules.sh -s x86_E5v2_Mellanox_GPU +source /scratch/nvarini/tensorflow-1.9.0-pip3/bin/activate + +module load gcc/6.4.0 cuda/9.1.85 openmpi/3.0.1-cuda python cudnn +export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH +srun python tf_cnn_benchmarks.py --batch_size=64 --model=resnet50 --num_gpus=1 --data_format=NCHW --variable_update=replicated --local_parameter_device=gpu --num_batches=3000 + +deactivate diff --git a/1.9-gpu/tensorflow-1.9.0-cp36-cp36m-linux_x86_64.whl b/1.9-gpu/tensorflow-1.9.0-cp36-cp36m-linux_x86_64.whl new file mode 100644 index 0000000..c082156 Binary files /dev/null and b/1.9-gpu/tensorflow-1.9.0-cp36-cp36m-linux_x86_64.whl differ