上篇博客https://blog.csdn.net/qq_38855907/article/details/104764936已经完成了cuda和cudnn的配置,接着就是开始配置tensorflow-gpu以及keras了。

开始前我们要明确需要做哪些工作

主要一点很关键 tensorflow-gpu、keras、python对应版本的选择!!!

keras配置tensorflow-gpu以及cuda和cudnn全过程!!!(下)

keras配置tensorflow-gpu以及cuda和cudnn全过程!!!(下)

当初在装的时候 我专门选择了已安装好的cud和cudnn版本 因此选择的是tensorflow-gpu1.5.0 python3.6 以及keras2.1.4

接下来开始一个个进行安装。

let's do it!

在Anaconda中创建TensorFlow的虚拟环境

右键以管理员身份运行Anacoda Prompt

在该终端输入下面这行代码,检测目前所创建的虚拟环境

conda info --envs 

然后创建虚拟环境:

conda create --name tensorflow-gpu python=3.6

3.7版本目前好像暂不支持,所以我选择3.6版本的python

然后=========

等创建完输入:

activate tensorflow-gpu

虚拟环境(python3.6)创建完成。

在虚拟环境中开始安装tensorflow-gpu1.5.0和keras2.1.4

因为是新创建的环境所以不存在tensorflow哈,如果在已存在的环境如base,可以先用pip uninstall tensorflow进行卸载

安装代码为pip install tensorflow-gpu==1.5.0

这里有一个小坑,如果无法正常安装要考虑两个原因了:

(1)是否以管理员身份运行anacoda prompt

  (2)将安装代码换成pip install --user tensorflow-gpu==1.5.0

安装完成后安装keras 代码:

pip install keras==2.1.4

不要用conda安装,因为conda安装时候会安装默认版本的packeges

安装完成后打开

keras配置tensorflow-gpu以及cuda和cudnn全过程!!!(下)

打开后选择Home->Appication on->tensorflow-gpu->install Spyder

keras配置tensorflow-gpu以及cuda和cudnn全过程!!!(下)

等下载好后 会出现

keras配置tensorflow-gpu以及cuda和cudnn全过程!!!(下)

 然后安装相应的包就可以使用GPU加速了!!!

希望能帮到你们呀!

相关文章:

  • 2021-11-07
  • 2021-06-01
  • 2021-12-05
  • 2021-06-24
  • 2021-09-30
  • 2021-05-10
  • 2021-08-04
  • 2021-04-12
猜你喜欢
  • 2021-04-05
  • 2021-04-24
  • 2021-06-30
  • 2021-11-13
  • 2022-01-21
  • 2021-09-17
  • 2021-08-21
相关资源
相似解决方案