win10如何简单安装tensorflow-gpu和cudnn

由于需求,我需要安装一个tensorflow-gpu 1.14的环境。
1,你要先安装anaconda,这个百度使用清华源资源的安装anaconda。
2,对anaconda添加清华源,打开cmd命令,输入以下命令,

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

3,搭建环境,创建一个tf1.14

conda creat -n tf1.14 python=3.7

4,激活环境

conda activate tf1.14

5,安装tensorflow-gpu1.14

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

此时还是无法运行tensorflow,因为没有安装cudatoolkit和cudnn,网上教程是去NVIDIA官网下载安装这些,但我试出一个快速安装运行的方法。
我当时在安装tensorflow2.0后,直接测试,居然可以运行,但安装1.14时无法运行,后来使用命令

conda list

查看2.0和1.14里的安装包有哪些不一样,发现1.14缺少cudatoolkit和cudnn两个包,于是用命令安装这两个包。

conda install cudatoolkit=10.0.130

然后开始从清华源读取包安装了

注意,后面一定要加10.0.130这个版本号,因为不加就默认安装最新版本,我当时没加就安装的是10.2版本,导致后面报错,没找到.dll文件。

再安装cudnn

conda install cudnn=7.6.5

完成后,可以运行代码测试了。

import tensorflow as tf  # 导入tensorflow
import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # 代码用于忽略级别 2 及以下的消息(级别 1 是提示,级别 2 是警告,级别 3 是错误)。
message = tf.constant("Welcome to network")  # 打印的信息是一个常量字符串
with tf.Session() as sess:
    print(sess.run(message).decode())  # 这里的输出结果是一个字节字符串。要删除字符串引号和“b”(表示字节,byte)只保留单引号内的内容,可以使用 decode() 方法。

结果为

Welcome to network
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值