CUDA HOME в установке pytorch

Question

Я установил Pytorch через Конда с 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: Двоичный файл nvcc не может быть расположен в вашем $ PATH. Либо добавьте его к своему пути, либо установите $ CUDAHOME

Я предполагаю, что мне нужно установить CUDAHOME в моем пути, но я не могу найти каталог cuda, имеющий двоичный файл nvcc. Есть какие-нибудь указатели на это?

Answer 1

Пакет CUDA, который распространяется через anaconda, не является полной установкой инструментария CUDA.Он включает только необходимые библиотеки и инструменты для поддержки numba и pyculib и других ускоряемых двоичных пакетов на GPU, которые они распространяют, например tensorflow и pytorch.

Если вам нужен полнофункциональный инструментарий CUDA (и, кажется, вы делаете), вам нужно будет установить один самостоятельно.Слово мудрому - установите ту же версию, что вы установили в Anaconda.С небольшим изменением PATH все должно работать.