pytorch安装中的CUDA HOME

我通过康达和cuda 7.5安装了pytorch

conda install pytorch=0.3.0 cuda75 -c pytorch

>>> import torch
>>> torch.cuda.is_available()
True

除此之外，我没有为cuda做任何其他安装，因为看起来pytorch与cuda一起提供

现在，我正在尝试设置yolo2

但是，

/make.sh

命令中出现错误

这就是错误所在

OSError:在$PATH中找不到nvcc二进制文件。要么将其添加到路径中，要么设置$CUDAHOME

---

我假设我需要在路径中设置CUDAHOME，但我无法找到任何包含nvcc二进制文件的cuda目录。有什么指示吗

通过anaconda分发的CUDA软件包不是完整的CUDA工具包安装。它只包括支持

numba

和

pyculib

以及它们分发的其他GPU加速二进制软件包（如

tensorflow

和

pytorch

）所需的库和工具

---

如果你需要一个功能齐全的CUDA工具包（看起来你需要），你需要自己安装一个。明智的话——安装与您在anaconda中安装的版本相同的版本。只要稍微修改一下路径，一切都会正常工作。

谢谢您的澄清。是的，我昨天也做了同样的事。分别安装了同一版本的CUDA和cuDNN，并为CUDA HOME设置了路径，并且运行正常。但是cuDNN也需要吗？