TensorFlow_GPU 1.15 + CUDA 10.0 + CUDNN 7.6 +Python 3.7 配置过程

   日期:2020-07-09     浏览:648    评论:0    
核心提示:TensorFlow_GPU 1.15 + CUDA 10.0 + CUDNN 7.6 +Python 3.7 配置过程主要是在conda下搭建GPU版本的TensorFlow,后续会提及Keras的安装TensorFlow_GPU 1.15的安装conda安装anaconda官网下载安装即可。创建conda环境创建一个新的环境,主要是为了万一出错只删除这个环境就行,不影响你原来的Python环境,如果你足够自信的话,可以不用创建 conda create -n 你

主要是在conda下搭建GPU版本的TensorFlow,后续会提及Keras的安装

TensorFlow_GPU 1.15的安装

conda安装

anaconda官网下载安装即可。

创建conda环境

创建一个新的环境,主要是为了万一出错只删除这个环境就行,不影响你原来的Python环境,如果你足够自信的话,可以不用创建

conda create -n "你的环境名" python = python版本号
创建环境代码示例 创建成功后提示进入和退出环境的用法
conda activate + 你的环境名

成功进入环境后,如下图所示

pip安装1.15版本的TensorFlow-GPU

使用阿里的源进行安装

pip install tensorflow-gpu==1.15 -i http://mirrors.aliyun.com/pypi/simple --trusted-host mirrors.aliyun.com

由于网络的原因,可能会失败,多试几次就可。

以下还有一些其他的国内源

pip  install  -i    [镜像源地址]    [安装包] 
例:pip  install  -i  https://pypi.tuna.tsinghua.edu.cn/simple  tensorflow-gpu

pypi的国内镜像源:
清华大学  https://pypi.tuna.tsinghua.edu.cn/simple/
阿里云  http://mirrors.aliyun.com/pypi/simple/
中国科技大学  https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban)  http://pypi.douban.com/simple/
中国科学技术大学  http://pypi.mirrors.ustc.edu.cn/simple/

CUDA 10.0的安装

首先确认自己的电脑最高支持什么版本的CUDA驱动

打开英伟达控制面板控制

点击右下角的“系统信息”,再点击“组件”,看到NVCUDA.DLL后面显示的版本,大于10.0即可(向下兼容)

到官网下载CUDA

https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

然后点击download,有点大,但是下载不慢

开始安装

双击下载好的可执行文件,解压目录什么的建议用默认的,虽然占了C盘,但是这样稳一点

解压要花几分钟,正常的,不要急。解压完成之后进入下图的界面

这里可能也会花一点时间,视电脑情况而定

随后,同意许可;安装选项选自定义,然后无脑下一步。

注意,这里的路径记一下,一会儿要用。

有的可能会提示没有支持版本的Visio studio,可以去官网下一个,也可以暂时忽略(使用过程中没有发现什么影响),直接勾选框框,然后next。

安装成功如图所示

CUDNN 7.6安装

cudnn的安装很简单,在网上找到对应版本下载,然后将下图中三个目录中的文件(注意:是文件夹下的文件!不是整个文件夹!也不是子文件夹!是文件!)复制到cuda安装目录(之前让记住的那个)对应的文件夹下即可

我的云盘有7.6版本的,其余版本的就要自己上网找了

http://101.201.142.67:12/cloud/index.php?share/file&user=100&sid=AE83XVTY

提取密码: Uloze

检验安装是否成功

进入新建的虚拟环境,然后输入Python,再输入如下代码

import tensorflow

出现成功加载动态库,即安装成功

ps:如果报错找不到cudart64_100.dll,重启一下电脑,再重复上述试验步骤即可。

Keras安装

先找与自己tensorflow版本对应的Keras

如图,我安装的是2.3.1

pip install keras==2.3.1  -i https://pypi.tuna.tsinghua.edu.cn/simple/

安装成功后,输入

import keras

出现如图所示字样,安装成功

在新的环境中安装Spyder

首先打开anaconda navigator

随后切换到新建的虚拟环境

然后点击Spyder的install(这里需要墙出去,如果没有墙,建议在命令行换国内源安装)

vscode选择python解释器

安装完成之后,直接在vscode中运行代码,还会用base环境,导致报错找不到tensorflow

需要将python解释器变为自己的新配置的虚拟环境

Ctrl+shift+p,然后输入python select interpreter选择解释器

选择自己新建的虚拟环境,再运行代码即可

或者

在命令行输入 conda activate tensorflow-gpu

又或者

点击左下角,然后选择解释器

如果以上教程还不能让你成功安装,建议使用conda一键安装的命令

conda install cudatoolkit=10.0
conda install cudnn=7.6
conda install tensorflow-gpu=1.15

ps:这种方法我没有试过,但有同学安装成功

 
打赏
 本文转载自:网络 
所有权利归属于原作者,如文章来源标示错误或侵犯了您的权利请联系微信13520258486
更多>最近资讯中心
更多>最新资讯中心
0相关评论

推荐图文
推荐资讯中心
点击排行
最新信息
新手指南
采购商服务
供应商服务
交易安全
关注我们
手机网站:
新浪微博:
微信关注:

13520258486

周一至周五 9:00-18:00
(其他时间联系在线客服)

24小时在线客服