在pytorch安装的CUDA HOME

Question

我用cuda 7.5通过conda安装了pytorch

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，但我无法找到任何具有nvcc二进制文件的cuda目录。有什么指针吗？

Answer 1

通过anaconda分发的CUDA包不是完整的CUDA工具包安装。它只包括必要的库和工具来支持numba和pyculib以及它们分发的其他GPU加速二进制包，如tensorflow和pytorch。

如果你需要一个功能齐全的CUDA工具包（你似乎也是这样），你需要自己安装一个。明智的话 - 安装在anaconda中安装的相同版本。只需稍加修改PATH，一切都应该正常工作。