机器学习环境配置:Anaconda+tensorflow GPU +CUDA+cuDNN+Keras

环境

工具版本号
anaconda
python3.5
tensorflow gpu1.15.0
CUDA10.0
cuDNN7.4
keras2.1.5

本篇主要是为了记录在安装tensorflow gpu,CUDA,cuDNN等过程中踩得坑,以及总结了其他博文中的一些内容。

首先要确定好版本号,先查看你本机的NVIDIA显卡驱动程序是否已经安装了CUDA,并确定CUDA的版本号,如果CUDA和tensorflow GPU 的版本号不对应的话,在安装过程中会产生一系列的问题,以下是我的总结,可以按照如下的安装步骤进行安装。

anaconda安装与环境配置

可以参考这篇博文:anaconda安装与环境变量

在安装tensorflow gpu,CUDA,cuDNN时主要是各个版本不兼容的问题。因此版本号的选择很重要。

一、确定本机是否支持CUDA
首先需要确定你本机的显卡是否支持安装Tensorflow GPU版本。
打开设备管理器查看显卡型号,谷歌搜索你的显卡型号;在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
本机支持CUDA

二、选择CUDA的版本号
CUDA下载
测试:
在控制台输入cncc -V
在这里插入图片描述
三、选择对应的cuDNN
cuDNN下载
cuDNN下载时需要登录,下载后将压缩包中对应的文件复制到CUDA安装目录下对应的文件中。
参考博文:cuDNN配置
四、选择对应版本的tensorflow gpu
版本号对应

tensorflow GPU

一、在anaconda中创建tensorflow环境
1、创建tensorflow环境

conda create -n tensorflow pip python=3.5  #虚拟环境的名称为tensorflow

2、激活tensorflow环境,并进入tensorflow环境下

activate tensorflow

3、下载tensorflow gpu 1.15.0版本

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.15.0

4、测试

1.进入tensorflow环境
>>activate tensorflow
2.输入python
>>python
3.输入以下命令
>>import tensorflow as tf
>>hello=tf.constant(‘hello,world’)
>>sess=tf.Session()
>>print(sess.run(hello))

在这里插入图片描述
显示以下,表示tensorflow安装成功。
在这里插入图片描述

Keras

pip install keras==2.1.5

测试:

import keras

没有报错表示安装成功。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝色Tu耳其

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值