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的根目录

win10安装CUDA CUDNN tensorflow-gpu

 

#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!
win10安装CUDA CUDNN tensorflow-gpu

 

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

上一篇:Ubuntu系统---安NVIDIA 驱动后 CUDA+cuDNN 安装


下一篇:程序员依然是这个时代,贫寒学子翻身的不二选择