win10安装CUDA CUDNN tensorflow-gpu

#1 安装anaconda

官网下载安装即可。python3.7版本

 

#2 安装CUDA CUDNN

注意tensorflow不同版本所需的CUDA及CUDA对应的CUDNN不同,注意版本匹配!

参考 https://blog.csdn.net/omodao1/article/details/83241074

https://developer.nvidia.com/cuda-gpus

https://developer.nvidia.com/cuda-toolkit-archive

https://developer.nvidia.com/rdp/cudnn-download

这里安装了CUDA10.0版本

快速安装即可

之后安装对应的CUDNN

官网下载安装包后解压,将得到的文件夹复制到CUDA10.0的根目录

 

#3. 安装tensorflow-gpu

打开Anaconda Prompt,进入Anaconda命令行管理界面。配置清华仓库镜,输入指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.7
新建一个名字叫“tensorflow-gpu”,python版本为3.的运行环境,此环境与Anaconda中其它环境隔离。输入“y“和回车后开始安装。

 

激活并进入环境,使后续指令在激活的环境中生效,输入指令:

conda activate tensorflow-gpu

 
升级pip到最新版,防止稍后的安装时,出现错误。输入指令:
python -m pip install --upgrade pip

 

安装tensorflow1.7.0及相应依赖包,输入指令:
pip install --ignore-installed --upgrade tensorflow-gpu
至此,tensorflow的安装完成。

 

#4 验证
在命令行中,先进入tensoflow-gpu,再进入python,并输入以下代码
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
输出
b’ Hello, TensorFlow!

 

参考https://blog.csdn.net/weixin_39290638/article/details/80045236

相关文章

MNIST数据集可以说是深度学习的入门,但是使用模型预测单张M...
1、新建tensorflow环境(1)打开anacondaprompt,输入命令行...
这篇文章主要介绍“张量tensor是什么”,在日常操作中,相信...
tensorflow中model.fit()用法model.fit()方法用于执行训练过...
https://blog.csdn.net/To_be_little/article/details/12443...
根据身高推测体重const$=require('jquery');const...