VS2015配置CUDN

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了VS2015配置CUDN相关的知识,希望对你有一定的参考价值。


  1. anaconda+vs+tensorflow-gpu+cudn+cudnn+python配置

vs2015(首先安装)

cuda9.0

cudnn7.0

tensorflow-gpu1.8.0

python3.6

参考:​​javascript:void(0)​​

2、could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

2018-10-29 17:01:58.920457: F T:\\src\\github\\tensorflow\\tensorflow\\core\\kernels\\conv_ops.cc:713] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms)

解决:

VS2015配置CUDN_github

在程序开头加入:

import tensorflow as tf

import os

os.environ["CUDA_VISIBLE_DEVICES"] = 0   #指定第一块GPU可用

config = tf.ConfigProto()

config.gpu_options.per_process_gpu_memory_fraction = 0.5  # 程序最多只能占用指定gpu50%的显存

config.gpu_options.allow_growth = True      #程序按需申请内存

sess = tf.Session(config = config)

以上是关于VS2015配置CUDN的主要内容,如果未能解决你的问题,请参考以下文章

win10+RTX3050ti+TensorFlow+cudn+cudnn配置深度学习环境

vs2015和emgu3.4.3可以配置吗

VS2015驱动开发环境配置

vs2015终于配置完成了

VS2015+Opencv3.2+QT5.8配置(一次配好)

vs 2015 + OPENGL 配置