Anaconda下安装tensorflow-gpu踩坑日记

装tensorflow-gpu需要CUDA(NVIDIA推出的通用并行计算架构,使GPU能够矩阵运算等),还有cuDNN(深度神经网络的GPU加速库)的支持。这里强调版本匹配!版本匹配!版本匹配! 以下放上三者对应版本:

在这里插入图片描述
注:之前到官网查看自己显卡与CUDA兼容性,是否有深度学习环境。

https://developer.nvidia.com/cuda-gpus#compute

我的是NoteBook GTX1060的
在这里插入图片描述

开始正式安装过程:

1、安装CUDA9.0 在这个网址查找CUDA已发布版本:

https://developer.nvidia.com/cuda-toolkit-archive

在这里插入图片描述
在这里插入图片描述

2、下载好CUDA Toolkit 9.0 后,我们开始下载cuDnn 7.0,需要注意的是,下载cuDNN需要在nvidia上注册账号,使用邮箱注册就可以免费的。登陆账号后才能下载。

https://developer.nvidia.com/rdp/cudnn-archive

在这里插入图片描述
在这里插入图片描述

3、安装CUDA9.0 toolkit和cuDnn7.0

在这里插入图片描述
打开cuda9.0安装包。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

接下来,解压“cudnn-9.0-windows10-x64-v7.zip”,将一下三个文件夹,拷贝到CUDA安装的根目录下。拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0与中间的文件夹合并。
在这里插入图片描述
下一步是配置系统环境变量,以下路径加入系统环境变量中:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

4、pip命令安装tensorflow-gpu,我用的是1.8.0版本(默认使用Anaconda安装,并且创建了tensorflow的虚拟python3.5.6环境,我用的anaconda的图形界面创的)

打开Anaconda Prompt
在这里插入图片描述
命令行下输入

activate tensorflow

使用清华镜像,快!

pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==1.8.0

验证tensorflow-gpu 1.8.0安装成功。

import tensorflow as tf
tf.test.gpu_device_name()

在这里插入图片描述

注:如果发生以下两个问题:

1.OSError: [WinError 126]找不到指定的模块。

2.ImportError: Could not find ‘cudart64_90.dll’. TensorFlowrequires that this DLL be installed in a directory that is named in your
%PATH%environment variable. Download and install CUDA 9.0 from this
URL: https://developer.nvidia.com/cuda-toolkit

分析:
1、一般是TensorFlow版本号与目前安装的cuda不搭配!在anaconda中输入conda list,查看TensorFlow版本号,如有不匹配重新卸载TensorFlow-gpu包,再次重装上合适版本的tensorflow-gpu包。
2、我的问题是在Anaconda Prompt中可以运行但是在VS code中报这个错。发现是编译器问题,就卸载VScode重装即解决了问题。

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值