获取CUDA_HOME环境路径 PYTORCH

10 人关注

我通过Anaconda在我的系统上安装了cuda，它有两个GPU，可以被我的python识别。

import torch
torch.cuda.is_available()
然而，当我试图通过其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，我不知道该设置哪个路径。
我试着寻找方法，但它给我返回了太多的cuda的路径。谁能帮我找到CUDA的路径？谢谢。


           
            
             你到底是如何找到你的安装目录的？


           
            
             @whitespace find / -type d -name cuda 2>/dev/null


         
          python


         
          deep-learning


         
          anaconda


         
          pytorch


          
           已采纳


          
           
            上述问题的解决方案!
           
           
            因为通过Anaconda安装的cuda并不是整个软件包。请从Nvidia网站手动安装cuda驱动[
            
             https://developer.nvidia.com/cuda-downloads
            
            ]
           
           
            安装完驱动后，pytorch就可以访问cuda路径了。你可以使用下面的示例代码测试cuda路径。问题解决了!!!
           
           
            检查安装。
           
           import os
print(os.environ.get('CUDA_PATH'))