win10安装tensorflow-gpu1.13.1+cuda10.0+cudnn7…

一，本机配置

Win10 64bit

NVIDIA GeForce GTX 960M

Python3.7（Anaconda）

二，安装CUDA

亲测，TensorFlow-gpu1.13.1支持cuda10.0的版本，所以我们可直接选择cuda10.0的版本

Window10下载CUDA10

安装步骤为：

1）运行.exe可执行和层序。安装软件会先运行一个系统检查，如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况，系统检查则会报错，无法进行下一步。如果报错了，就最先考虑下是不是自己的显卡不被该版本支持。在安装过程中，它会先自动安装和你电脑对应的NVIDIA驱动。
2）选择默认安装就行。

三，安装cuDNN

cuDNN7.3.1下载

下载cuDNN需要自己先在以上网址（cuDNN官网）注册个账号，注册成功后登陆进去选择cuDNN7.3.1下载即可。

下载完成后是一个.zip包，解压后进入目录，会发现三个子目录/bin，/include，/lib/x64。进去后会发现有三个文件，将这三个文件分别拷贝到cuda的安装目录的对应子目录下

默认cuda的安装目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

四，配置系统Path环境变量

检查你的系统Path环境变量中是否有这四项，没有请手动添加

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

五，安装tensorflow-gpu

打开Anaconda Prompt，输入

pip install tensorflow-gpu

会自动下载并安装tensorflow-gpu

六，安装测试

打开pycharm，创建一个test.py文件，输入

import tensorflow as tf

tf.enable_eager_execution()
print(tf.__version__)
hello = tf.constant("hello tensorflow")
print(hello.numpy())

出现下列结果表明安装成功

原文链接:https://www.cnblogs.com/SystemCall/p/10660077.html
如有疑问请与原作者联系

标签：

版权申明：本站文章部分自网络，如有侵权，请联系：west999com@outlook.com
特别注意：本站所有转载文章言论不代表本站观点，本站所提供的摄影照片，插画，设计作品，如需使用，请与原作者联系，版权归原作者所有