Modules needed to run this code on Cori:
- PrgEnv-llvm/12.0.0-git_20210117
- cudnn/8.0.5
- cgpu
- cuda/11.1.1
- gcc/8.3.0
- pytorch/1.8.0-gpu
To compile and run the code on a GPU, use:
- mkdir build
- cd build/
- CXX=/global/homes/n/namehta4/kokkos/install_pm_llvm/bin/nvcc_wrapper cmake -DCMAKE_PREFIX_PATH="/global/homes/n/namehta4/kokkos/install_pm_llvm" -DKok kos_ROOT=/global/homes/n/namehta4/kokkos/install_pm_llvm ../
- make -j
- srun -n 1 ./FirstNN'