相关疑难解决方法(0)

pytorch安装中的CUDA HOME

我通过conda和cuda 7.5安装了pytorch

conda install pytorch=0.3.0 cuda75 -c pytorch

>>> import torch
>>> torch.cuda.is_available()
True
Run Code Online (Sandbox Code Playgroud)

除此之外,我没有为cuda做任何其他安装,因为pytorch似乎是cuda附带的

现在,我正在尝试设置yolo2 https://github.com/longcw/yolo2-pytorch

但是,我在./make.sh命令中遇到错误

这是错误

OSError:nvcc二进制文件找不到您的$ PATH中。将其添加到您的路径,或设置$ CUDAHOME

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

cuda pytorch yolo

1
推荐指数
1
解决办法
2264
查看次数

标签 统计

cuda ×1

pytorch ×1

yolo ×1