microsoft
/
CNTK
зеркало из
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Add a comment about how to get nvml.h

This commit is contained in:
Yu 2015-05-24 13:34:53 -04:00
Родитель 6905080482
Коммит 33a6902b46
1 изменённых файлов: 3 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

5
Makefile.gpu
Просмотреть файл

@ -24,6 +24,7 @@
# You may need to do the following or something similar for all this to work # 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 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 # 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 CC = g++-4.8
NVCC = nvcc NVCC = nvcc
@ -62,11 +63,11 @@ else
MATHLIB_DEFINE = -DUSE_ACML MATHLIB_DEFINE = -DUSE_ACML
endif endif
CUDA_PATH = /usr/local/cuda-7.0 CUDA_PATH = /scratch/cuda-7.0
CUDA_INCLUDE = $(CUDA_PATH)/include CUDA_INCLUDE = $(CUDA_PATH)/include
CUDA_LIB = -L$(CUDA_PATH)/lib64 -lcublas -lcudart -lcuda -lcurand -lcusparse -lnvidia-ml 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) INCFLAGS = -I Common/Include -I Math/Math -I MachineLearning/CNTK -I $(CUDA_INCLUDE) -I $(MATHLIB_INCLUDE) -I $(NVML_INCLUDE)