pytorch 安装中的 CUDA HOME

Posted 2023-02-16

【中文标题】pytorch 安装中的 CUDA HOME

【英文标题】：CUDA HOME in pytorch installation

【发布时间】：2019-02-17 06:29:12

【问题描述】：

我通过 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 目录。有什么指示吗？

【参考方案1】：

通过 anaconda 分发的 CUDA 包不是完整的 CUDA 工具包安装。它仅包含必要的库和工具来支持numba 和pyculib 以及它们分发的其他GPU 加速二进制包，例如tensorflow 和pytorch。

如果您需要一个功能齐全的 CUDA 工具包（您似乎确实需要），您需要自己安装一个。明智的话——安装与您在 anaconda 中安装的版本相同的版本。稍加修改 PATH，一切都应该正常工作。

【讨论】：

感谢您的澄清。是的，我昨天也做了同样的事情。分别安装了相同版本的CUDA，cuDNN以及设置CUDA HOME的路径，它工作。但是否也需要 cuDNN？