diff --git a/etc/env/raven/cuda b/etc/env/raven/cuda new file mode 100644 index 0000000..7e07ce6 --- /dev/null +++ b/etc/env/raven/cuda @@ -0,0 +1,56 @@ +mods=( + cuda/11.6 + intel/19.1.2 + mkl/2020.4 + impi/2019.8 + autoconf/2.69 + automake/1.15 + libtool/2.4.6 +) + + +module purge +module load ${mods[@]} +LIB_PATH="${CUDA_HOME}/lib64" +export CUDA_ROOT=${CUDA_HOME} +export CUDA_LDFLAGS="-L${LIB_PATH} -lcuda -L${LIB_PATH} -lcudart -L${LIB_PATH} -lcublas" +export CUDA_CXXFLAGS="-I${CUDA_HOME}/include" + +export LD_LIBRARY_PATH="${MKL_HOME}/lib/intel64_lin:${LD_LIBRARY_PATH}" + +BLAS_STATIC_PATH="$MKL_HOME/lib/intel64/libmkl_intel_lp64.a" + +ls ${LIB_PATH}/libcublas.so +ls ${LIB_PATH}/libcudart.so + +cat <