From 33a6902b46bd1319732c57a8023e212c3a33235e Mon Sep 17 00:00:00 2001 From: Yu Date: Sun, 24 May 2015 13:34:53 -0400 Subject: [PATCH] Add a comment about how to get nvml.h --- Makefile.gpu | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/Makefile.gpu b/Makefile.gpu index d42fbc8c2..ab1f987ac 100644 --- a/Makefile.gpu +++ b/Makefile.gpu @@ -24,6 +24,7 @@ # You may need to do the following or something similar for all this to work # export LD_LIBRARY_PATH=/usr/local/acml5.3.0/gfortran64/lib:/usr/local/cuda/lib64:/usr/local/lib # export PATH=$PATH:/usr/local/bin:/usr/local/cuda/bin +# Install the GPU deployment Kit@ https://developer.nvidia.com/gpu-deployment-kit CC = g++-4.8 NVCC = nvcc @@ -62,11 +63,11 @@ else MATHLIB_DEFINE = -DUSE_ACML endif -CUDA_PATH = /usr/local/cuda-7.0 +CUDA_PATH = /scratch/cuda-7.0 CUDA_INCLUDE = $(CUDA_PATH)/include CUDA_LIB = -L$(CUDA_PATH)/lib64 -lcublas -lcudart -lcuda -lcurand -lcusparse -lnvidia-ml -NVML_INCLUDE = /usr/include/nvidia/gdk +NVML_INCLUDE = /scratch/usr/include/nvidia/gdk # install and include it @ https://developer.nvidia.com/gpu-deployment-kit INCFLAGS = -I Common/Include -I Math/Math -I MachineLearning/CNTK -I $(CUDA_INCLUDE) -I $(MATHLIB_INCLUDE) -I $(NVML_INCLUDE)