CUDA HOME в установке pytorch
Я установил pytorch через conda с cuda 7.5
conda install pytorch=0.3.0 cuda75 -c pytorch
>>> import torch
>>> torch.cuda.is_available()
True
Я не делал никаких других установок для cuda, кроме этой, так как похоже, что pytorch поставляется с cuda
Теперь я пытаюсь настроить yolo2 https://github.com/longcw/yolo2-pytorch
Однако я получаю сообщение об ошибке в команде ./make.sh.
это ошибка
OSError: The nvcc binary could not be located in your $PATH. Either add it to your path, or set $CUDAHOME
Я предполагаю, что мне нужно установить CUDAHOME на моем пути, но я не могу найти какой-либо каталог cuda, имеющий двоичный файл nvcc. Есть указатели на это?
Ответы 1
Пакет CUDA, который распространяется через anaconda, не является полной установкой набора инструментов CUDA. Он включает только необходимые библиотеки и инструменты для поддержки numba и pyculib и других распространяемых ими двоичных пакетов с ускорением на GPU, таких как tensorflow и pytorch.
Если вам нужен полнофункциональный инструментарий CUDA (а кажется, что вам нужен), вам нужно будет установить его самостоятельно. Мудрое слово - установите ту же версию, что и в Anaconda. После небольшой модификации PATH все должно работать.
Спасибо за разъяснения. Да, вчера я сделал то же самое. Установил ту же версию CUDA отдельно, а также cuDNN и установил путь для CUDA HOME, и это сработало. Но нужен ли был cuDNN?