我通过 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 的路径吗?提前致谢。
最佳答案
以上问题的解决方案!
由于通过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
关于python - 获取CUDA_HOME环境路径PYTORCH,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52731782/