使用CUDA遇到的坑

在使用PaddlePytorch或者是Tensorflow的时候,可能出现以下错误

ExternalError: CUDA error(100), no CUDA-capable device is detected. 
      [Hint: 'cudaErrorNoDevice'. This indicates that no CUDA-capable devices were detected by the installed CUDA driver. ] (at ..\paddle\fluid\platform\gpu_info.cc:108)
      [operator < fill_constant > error]

或者是

Compiled with WITH_GPU, but no GPU found in runtime.

说白了就是GPU不可用,研究了很久很久,尝试了各种方法,最后找到了解决方案:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'

只需要在代码中添加以上代码即可,需要说明的是,一定要加在文件的最顶部,再引入其他的深度学习库!

以下是正确的使用方法:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
import paddle

以下是错误的使用方法:

import paddle
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
  • 4
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值