使用CUDA遇到的坑

在使用Paddle、Pytorch或者是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'

使用CUDA遇到的坑相关推荐

ubuntu 16.04安装nVidia显卡驱动和cuda/cudnn踩坑过程
推荐新版安装教程 http://blog.csdn.net/chenhaifeng2016/article/details/78874883 安装深度学习框架需要使用cuda/cudnn(GPU)来加 ...
cudnn 安装失败_Win10下安装tensorflow环境的一些坑
2020更新: Attention!由于tensorflow更新频繁,特别是现在的2.x版本,改动较大,以下内容是基于tf-1.14版本的.本文基本内容如下: 导入Numpy报错问题解决 CUDA,c ...
双系统Ubuntu22.04深度学习环境配置与踩坑记录
双系统Ubuntu22.04深度学习环境配置踩坑记录前言目录相关版本主要参考教程 Ubuntu安装 Nvidia和CUDA安装踩坑经历官网安装所遇问题 cuDNN安装 Anaconda安装 ...
ubuntu16.04 安装cuda9.0+cudnn7.0.5+tensorflow+nvidia-docker配置GPU服务
[摘要] docker很好用,但是在GPU服务器上使用docker却比较复杂,需要一些技巧,下面将介绍一下在ubuntu16.04环境下的GPU-docker环境搭建过程. 第一步: 删除之前的nvi ...
win10+tensorflow-gpu+pycharm+anaconda...的成功安装，顺利在spyder import tensorflow
装了好几天的tensorflow,终于今天装好了.在这里记录一下,免得下次遇到问题忘了怎么解决. 首先我是找了几篇文章,将需要用上的环境都准备好. 第一篇参考文章,不过这篇文章少了前期几个步骤. 第 ...
在服务器上安装tensorflow-gpu版本及其使用
[Deprecated]在服务器上安装tensorflow-gpu版本及其使用已过时,不建议参考本文档是在服务器上,不要su权限,全部命令行下载安装配置教程 ###### 系统 Centos7 显 ...
TensorFlow 中的 LRNOp
TensorFlow 中的 LRNOp 与 Caffe 的差异: 直接使用平方和而不是像论文中一样使用平方和的均值,因此算子的推荐参数有所不同: 仅支持 NHWC 格式的输入: CPU 后端除框架源码 ...
从头开始训练一个检测QR二维码区域的YOLOv3模型
条形码和二维码在识别的时候主要包含定位和解码两个步骤.寻找码的位置,除了用传统的图像算法之外,也可以借助深度学习.那么深度学习的效率如何,我做了一个实验. 为QR二维码训练YOLOv3模型编译Dar ...
pytorch离线快速安装
1.pytorch官网查看cuda版本对应的torch和torchvisionde 版本(ncvv -V,nvidia-sim查看cuda对应的版本) 2.离线下载对应版本,网址https://dow ...

使用CUDA遇到的坑

使用CUDA遇到的坑相关推荐

最新文章

热门文章