码迷,mamicode.com
首页 >  
搜索关键字:cudnn    ( 244个结果
记一次 Ubuntu 下 NVIDIA 驱动 + CUDA + CUDNN 的卸载重装爬坑之旅
因为一些意外,Ubuntu 中的 nvidia 驱动被卸载了。。。 于是我就开始了一个漫长的爬坑之旅。。。这里记录一下过程 我的Ubuntu版本是 18.04 按住ctrl+alt+ f3 切换到纯命令行模式 使用以下命令进一步卸载干净 sudo apt-get --purge remove nvi ...
分类:系统相关   时间:2020-03-09 17:52:21    阅读次数:117
在windows上极简安装GPU版AI框架
在windows上极简安装GPU版AI框架如果我们想在windows系统上安装GPU版本的AI框架,比如GPU版本的tesnorflow,通常我们会看到类似下面的安装教程官方版本安装CUDA安装cuDNN配置环境变量安装python环境安装gpu版的tensorflow开发包咋看上去好像不是很复杂,但是其中坑多到你怀疑人生。下载的cuDNN时候需要注册,而且因为cuDNN文件在外网,下载速度很慢。
分类:Windows程序   时间:2020-03-07 22:37:03    阅读次数:94
在windows上极简安装GPU版AI框架(Tensorflow、Pytorch)
在windows上极简安装GPU版AI框架 如果我们想在windows系统上安装GPU版本的AI框架,比如GPU版本的tesnorflow,通常我们会看到类似下面的安装教程 官方版本 安装CUDA 安装cuDNN 配置环境变量 安装python环境 安装gpu版的tensorflow开发包 咋看上去 ...
分类:Windows程序   时间:2020-03-07 20:33:25    阅读次数:97
pytorch gpu~ cuda cudacnn安装是否成功的测试代码
# CUDA TEST import torch x = torch.Tensor([1.0]) xx = x.cuda() print(xx) # CUDNN TEST from torch.backends import cudnn print(cudnn.is_acceptable(xx))# ...
分类:其他好文   时间:2020-03-05 01:17:50    阅读次数:228
模型训练错误:Failed to get convolution algorithm,This is probably because cuDNN failed to initialize.
一、问题: 最近在利用keras训练模型的时候出现以下错误,开始一看以为是cuDNN的版本不对,更换后还是没有效果,后面在网上查找后发现是显存的问题,更改为动态分配内存就可以了。 二、解决方法: 在训练的脚本开头添加以下代码进行动态分配内存,然后重新运行就可以。 import tensorflow ...
分类:其他好文   时间:2020-03-01 10:26:39    阅读次数:55
去除警告: FutureWarning: In future, it will be treated as `np.float64 == np.dtype(float).type`.
文章目录: 一、警告原因分析 二、解决方法 最近在使用tensorflow gpu的时候总是报警告,就我这强迫症能够忍他吗,no!绝对不能忍!!! 我的环境: cudnn=7.0 cuda=9.0 tensorflow gpu=1.8 或 1.9 或 1.10.0 或 1.11.0 在上面的这个环境 ...
分类:其他好文   时间:2020-02-21 10:02:00    阅读次数:116
NVCC Makefile 示例
GPU=0 CUDNN=0 OPENCV=0 OPENMP=0 DEBUG=0 ARCH= -gencode arch=compute_30,code=sm_30 \ -gencode arch=compute_35,code=sm_35 \ -gencode arch=compute_50,cod ...
分类:其他好文   时间:2020-02-13 00:20:54    阅读次数:116
tensorflow-gpu与cuda 关系
对于版本号大于1.13的tensorflow gpu版本,如1.14、1.15和2.0,要安装CUDA10.0,不要安装最新的CUDA10.1,安装后会提示缺少很多库文件,而导致GPU版本的tensorflow无法使用。 CUdnn与CUDA的对应关系 NVIDIA官网链接:https://deve ...
分类:其他好文   时间:2020-02-06 22:53:51    阅读次数:119
OpenCV配置cuda
以openCV4.2配置cuda10.2为例。总共参考三个:本篇博客、CMake编译OpenCV、https://www.bilibili.com/video/av71643385就可以配置完成。 【下载】 CUDA Toolkit和cuDNN https://developer.nvidia.co ...
分类:其他好文   时间:2020-02-05 00:05:54    阅读次数:95
anaconda python3.7 安装 tensorflow-gpu 2.0.0 beta1 配置PyCharm
参考tensorflow 公众号《tensorflow2.0 安装指南》 https://mp.weixin.qq.com/s/7rNXFEC5HYe91RJ0-9CKdQ 1. NVIDIA驱动程序安装 安装对应的CUDA 和 cudnn (在tensorflow 公众号《tensorflow2. ...
分类:编程语言   时间:2020-02-02 18:01:57    阅读次数:156
244条   上一页 1 ... 3 4 5 6 7 ... 25 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!