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