Убедитесь, что CuDNN установлен / работает - mac
Недавно я установил графический процессор на свой Mac, и теперь я пытаюсь заставить графический процессор выполнять тяжелую работу по обучению моих нейронных сетей.
Есть ли команда терминала для проверки, был ли cuDNN установлен на Mac? Я использую инфраструктуру тензорного потока, поэтому я предполагаю, что эта команда caffe не работает (я могу связать исходный код в необходимом стеке, поэтому я не буду публиковать две ссылки, потому что я слишком новый)
/usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
Я установил cuDNN, используя следующее:
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow_gpu-0.12.1-py3-none-any.whl
pip install --upgrade $TF_BINARY_URL
export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH="$CUDA_HOME/lib:$CUDA_HOME:$CUDA_HOME/extras/CUPTI/lib"
export LD_LIBRARY_PATH=$DYLD_LIBRARY_PATH
Который по существу копирует файлы cuDNN в папку cuda. Вот как выглядят мои каталоги в папке cuda (в случае, если это актуальная информация): cuda / lib / cuda / include /
Ask: Терминальная команда, чтобы проверить, установлен ли Cuda на самом деле.