【深度学习】windows下cuda+cudnn+anaconda+tensorflow-gpu环境搭建

穿精又带淫゛_ 提交于 2019-12-02 01:32:19

1、在这里查看支持的做机器学习计算的显卡:

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

需要到NVIDIA下载cuda与cudann,这里最重要的是注意cuda与cudann与tensorflow三者的搭配

 

2、关于nvidia显卡驱动版本,cuda版本,cudnn版本关系参考:

https://blog.csdn.net/iamqianrenzhan/article/details/89343601

https://blog.csdn.net/u013066730/article/details/80980940

查看的电脑上的NVIDIA控制面板里的系统信息窗口里的组件信息,显示的NVIDIA版本是是cuda driver版本,并非要安装的cuda  runtime的版本

3、安装cuda

官方指南:

https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

csdn博客:

https://blog.csdn.net/XCCCCZ/article/details/80385448

https://blog.csdn.net/u010618587/article/details/82940528

4、下载cuDNN

官方指南:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installwindows

解压cudnn,将里面的lib,include,bin目录里面的文件,拷贝至cuda安装文件对应的lib,include,bin目录中, cuda的系统默认安装对应为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\

5、安装anaconda及创建环境

https://blog.csdn.net/ITLearnHall/article/details/81708148

Anaconda下安装tensorflow-gpu  keras-gpu:

https://blog.csdn.net/qq_33039859/article/details/77412781

指定tensorflow版本,conda install tensorflow-gpu=1.12(1.12为笔者需要)

6、查看gpu状态

添加C:\Program Files\NVIDIA Corporation\NVSMI至系统环境变量,在cmd命令行输入nvidia-smi查看

 

 

 

 

 

 

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!