在windows上安装tensorflow gpu

近期,我在电脑上安装了tensorflow gpu版本,遇到一些问题,在这里写一些解决方案。

1、安装python3

2、安装tensorflow gpu:pip install tensorflow-gpu

3、查看显示适配器中关于NVIDIA,https://developer.nvidia.com/cuda-gpus,查看是否支持cuda

4、下载cuda,https://developer.nvidia.com/cuda-downloads,根据我的安装,发现tensorflow只支持cuda8.0

5、下载cudnn v5.1,cudnnv6.0安装完后会报错,将cudnn中lib,include,bin中文件移到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 ,可以根据你的路径变换。

6、测试,打开python,输入import tensorflow as tf,如无报错,则安装成功

7、如果报错 No module named '_pywrap_tensorflow_internal',则是因为缺少msvcp140.dll,需要安装visual c++ 2015 redistributable Package,https://www.microsoft.com/en-us/download/details.aspx?id=52685

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值