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.

如果通过cuda的方法安装了cuda，则

https://lfd.readthedocs.io/en/latest/install_gpu.html页将提供有关设置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

获取CUDA_HOME环境路径PYTORCH

1 个答案: