[深度学习]OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

Solution to above issue!

As cuda installed through anaconda is not the entire package. Please install cuda drivers manually from Nvidia Website [ https://developer.nvidia.com/cuda-downloads ]

After installation of drivers, pytorch would be able to access the cuda path. You can test the cuda path using below sample code. Problem resolved!!!

CHECK INSTALLATION:

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

OUTPUT:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
报错提示"OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root."表明CUDA_HOME环境变量未设置,需要将其设置为CUDA安装的根目录。 你可以按照以下方法解决该问题: 1. 临时解决方法:在命令行中执行命令"export CUDA_HOME=/usr/local/cuda-X.X",将CUDA_HOME设置为你的CUDA安装根目录的路径。请将X.X替换为你实际的CUDA版本号。 2. 永久解决方法:编辑 ~/.bashrc 文件,可以使用命令"sudo vi ~/.bashrc"打开文件进行编辑。在文件中添加一行代码"export CUDA_HOME=/usr/local/cuda-X.X",并将X.X替换为你实际的CUDA版本号。保存文件并退出。这样每次打开终端时,CUDA_HOME环境变量都会被自动设置。 通过以上方法设置CUDA_HOME环境变量后,你应该能够解决该报错并继续使用相关的CUDA功能。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [复原深度补全网络 ACMNet 填坑记录](https://blog.csdn.net/weixin_44832383/article/details/121289383)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [报错 OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.](https://blog.csdn.net/OOFFrankDura/article/details/113632416)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值