Add a comment about how to get nvml.h
This commit is contained in:
Родитель
6905080482
Коммит
33a6902b46
|
@ -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)
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче