获取 CUDA_HOME 环境路径 PYTORCH

Question

我在我的系统上通过 anaconda 安装了 cuda，该系统有 2 个 GPU，我的 python 可以识别这些 GPU。

import torch
torch.cuda.is_available()
true

但是，当我尝试通过其 C API 运行模型时，我收到以下错误：

~/anaconda3/lib/python3.6/site-packages/torch/utils/cpp_extension.py in _join_cuda_home(*paths)
722     '''
723     if CUDA_HOME is None:
--> 724         raise EnvironmentError('CUDA_HOME environment variable is not set. '
725                                'Please set it to your CUDA install root.')
726     return os.path.join(CUDA_HOME, *paths)

OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

https://lfd.readthedocs.io/en/latest/install_gpu.html 页面给出了设置 CUDA_HOME 路径的说明，如果通过他们的方法安装了 cuda。由于我通过 anaconda 安装了 cuda，我不知道要设置哪个路径。

我尝试了 find 方法，但它返回的 cuda 路径太多。有人可以帮我走上 CUDA 的道路吗？提前致谢。

Answer 1

以上问题的解决方案！

由于通过 anaconda 安装的 cuda 不是整个软件包。请从 Nvidia 网站手动安装 cuda 驱动程序[https://developer.nvidia.com/cuda-downloads]

安装驱动后，pytorch 将能够访问 cuda 路径。您可以使用以下示例代码测试 cuda 路径。问题解决了！！！

检查安装：

import os
print(os.environ.get('CUDA_PATH'))

输出：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1