库达之家在 pytorch 安装

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 进行一点点修改，一切应该都可以正常工作。