tensorflow——记录装多版本GPU的过程

写在前面

我之前按照keras中文文档已经安装好了tenforflow-gpu 2.1 ,并且在这个基础上安装好了keras,但是在github找的许多代码都是基于tensorflow-gpu 1.×的keras框架,同时我又不想覆盖掉我的tensorflow-gpu 2.1,并且我之前安装的cuda版本为10.1。

1. 安装要求

tensorflow1.8-gpu
cuda9.0
cudnn7.0
在这里插入图片描述

2. 安装过程

打开你的Anaconda Prompt,开始操作起来吧

  • 1.创建一个新的虚拟环境,保证包依赖的独立,互不影响
# 一定要注意python的版本,否则会报错
conda create -n env_tf1.8_gpu python=3.6
  • 2.进入该虚拟环境
conda activate env_tf1.8_gpu
  • 3.下载cuda,版本为9.0
conda install cudatoolkit=9.0
  • 4.下载cudnn,版本为7.0
    下载cudnn稍微麻烦一点,如果使用conda的话不能直接下载,会出现找不到Package的错误,我们从官网上直接下载,如果没注册得注册一下
    官网下载cudnn
    下载完之后,里边有个cuda文件夹,把它解压出来,里边有三个文件夹分别是:bin,include,lib,剩下哪个txt文件不用管。
    在这里插入图片描述
    在这里插入图片描述
    接下来这步至关重要:
    将三个文件夹下得文件分别放入该虚拟环境同名文件夹下,举个例子,解压出来得bin文件夹,打开后有一个文件如下图所示
    在这里插入图片描述
    然后进入到你的虚拟环境目录下,找到bin目录,将上面得文件放入这里,以此类推
    在这里插入图片描述

  • 5.下载tensorflow1.8-gpu

pip install tensorflow-gpu==1.8.0

-6.测试

1. conda activate env_tf1.8_gpu # 进入虚拟环境
2. python # 进入python得shell模式
3. import tensorflow as tf
4. tf.__version__() # 检查版本是否正确
5. tf.test.is_gpu_available() # 检查是否gpu已经启动,若返回True,则成功
  • 7.安装成功
  • 8.安装心得
    任何版本得gpu版本都能按照上述完成安装
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

InceptionZ

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

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

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

打赏作者

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

抵扣说明:

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

余额充值