python - 获取CUDA_HOME环境路径PYTORCH

Question

我通过 anaconda 在我的系统上安装了 cuda，该系统有 2 个 GPU，它被我的 python 识别。

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，页面给出了设置 CUDA_HOME 路径的说明。由于我已经通过 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