Anaconda镜像安装tensorflow-gpu1.14及Keras超详细版

一、CUDA和cuDNN的安装及环境配置

https://developer.nvidia.com/cuda-toolkit-archiv
首先从该网页下载tensorflow对应版本的的CUDA,tensorflow1.14可以使用CUDA10.0cuDNN7.6.4下载好后按要求进行安装。
https://developer.nvidia.com/cudnn
然后在这个网站下载cuDNN,解压后依次复制到CUDA的安装路径中,路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

Tensorflow对应的Python、Cuda、cuDNN版本:
Tensorflow对应的Python、Cuda和cuDNN版本
下载安装好后,需要在环境变量中添加路径。
PATH中手动添加:

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

前两个一般安装好CUDA后会自动配置好。(注意:若没有显卡驱动,需要先到英伟达官网下载最新的显卡驱动)

二、安装Anaconda

https://www.anaconda.com/download/
官网下载Anaconda后,按照其他博客安装好。
详细安装链接:Anaconda、Cuda、cuDNN
在开始栏找到Anaconda,打开Anaconda Prompt,进入终端后方便镜像下载。Anaconda开始栏所在位置

0.导入镜像包
设置额外的源,直接从速度快的源进行下载。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/

参考链接:https://blog.csdn.net/feifei3211/article/details/80361227
1.创建Python环境

conda create --name tensorflow-gpu python=3.6  

代表创建一个python3.6的环境,我们把它命名为tensorflow-gpu

2.激活环境

conda activate tensorflow-gpu

激活环境,方便在该环境中安装库

3.安装tensorflow-gpu的1.14版本
首先升级pip,防止出现问题

python -m pip install  -i https://pypi.doubanio.com/simple/ --upgrade pip
python -m pip install  -i https://pypi.doubanio.com/simple/ --upgrade setuptools

然后镜像安装tensorflow-gpu

# pip install -i https://pypi.doubanio.com/simple/ tensorflow-gpu==1.14
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn tensorflow-gpu==1.14

4.训练测试代码
输入

python

进入到python内部,输入以下代码测试tensorflow是否安装成功

import tensorflow as tf
sess = tf.Session()
a = tf.constant(1)
b = tf.constant(2)
print(sess.run(a+b))

退出python返回虚拟环境中

exit()

5.镜像安装Tensorflow对应Keras2.2.5版本

pip install -i https://pypi.doubanio.com/simple/ keras==2.2.5

其他库的安装也可以使用此行代码,只需更改库名即可。
Tensorflow对应的Keras版本:
Tensorflow与Keras版本对应
6.安装其他库

pip install -i https://pypi.doubanio.com/simple/ opencv-python
pip install -i https://pypi.doubanio.com/simple/ pillow
pip install -i https://pypi.doubanio.com/simple/ matplotlib

7.安装Spyder编译器

conda install spyder

如果安装失败,则重新执行此行命令。多次安装均失败,则用记事本打开C:\Users\电脑用户名路径下的.condarc,删掉 - defaults,再重新执行此行代码。

8.所有需要的库安装好后,训练模型

  • 10
    点赞
  • 33
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值