pytorch 安装中的 CUDA HOME

Question

我通过 conda 和 cuda 7.5 安装了 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，一切都应该正常工作。