Skip to content
Snippets Groups Projects
Unverified Commit 8c1f0b13 authored by Luca Capello's avatar Luca Capello
Browse files

t/tensorflow/hello/testTensorFlow_1.7.0.sh: new file

parent 4b6a15a0
No related branches found
No related tags found
No related merge requests found
#!/bin/sh
#SBATCH --cpus-per-task=1
#SBATCH --job-name=testTensorFlow
#SBATCH --ntasks=1
#SBATCH --time=05:00
#SBATCH --output=slurm-%J.out
#SBATCH --gres=gpu:1
#SBATCH --constraint="V5|V6"
#SBATCH --partition=shared-gpu
## TensorFlow
module load GCC/6.4.0-2.28 OpenMPI/2.1.2 TensorFlow/1.7.0-Python-3.6.4
## CUDA
module load cuDNN/7.0.5-CUDA-9.1.85
srun python helloworld.py
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment