您的当前位置:首页正文

CUDA+cuDNN+Tensorflow-GPU Instal

来源:华拓网

安装完以后运行nvcc -V命令确保安装成功

打开 C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0 并双击 Samples_vs2015.sln 文件用VS加载,然后build solution, 可能会出现一些failed不过没关系。

在CUDA Samples文件夹找到下图中路径执行devicequery命令。官方文档有两处错误,一是说同时提供源码和binary, 其实只有源码所以需要用VS编译,二是说执行文件在release文件夹里,其实在debug目录下。

最后打开Anaconda命令行窗口,执行pip install --upgrade tensorflow-gpu

安装完后执行以下官方给的这段代码能正常执行就应该是安装成功了

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

在命令行以交互方式执行也可以看到GPU的信息

虽然这里有一个AVX2的warning, 不过没有关系,stackoverflow说: If you have a GPU, you shouldn't care about AVX support, because most expensive ops will be dispatched on a GPU device (unless explicitly set not to).