pytorch安装中的CUDA HOME

时间:2018-09-12 14:56:29

标签: cuda pytorch yolo

我使用cuda 7.5通过conda安装了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:无法在$ PATH中找到nvcc二进制文件。将其添加到您的路径,或设置$ CUDAHOME

我假设我需要在路径中设置CUDAHOME,但是我无法找到任何具有nvcc二进制文件的cuda目录。有任何指针吗?

0 个答案:

没有答案