cuda - 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 目录。有什么指示吗？

标签： cudapytorchyolo

Answer 1

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

如果您需要一个功能齐全的 CUDA 工具包（您似乎确实需要），您需要自己安装一个。明智的说法——安装与您在 anaconda 中安装的版本相同的版本。通过一点点 PATH 修改，一切都应该正常工作。