配置深度学习GPU环境

个人博客,欢迎来撩 fangzengye.com

安装CUDA

https://developer.nvidia.com/cuda-90-download-archive

根据自己 版本下载

安装cuDNN

  1. https://developer.nvidia.com/cudnn上下载cudnn相应版本的压缩包(可能需要注册或登录)。

  2. 如果这个压缩包不是.tgz格式的,把这个压缩包重命名为.tgz格式。解压当前的.tgz格式的软件包到系统中的任意路径,解压后的文件夹名为cuda,文件夹中包含三个文件夹:一个为include,另一个为lib64,还有一个是bin,然后复制到CUDA_PATH下面。

  3. 将解压后的文件中的lib/x64文件夹关联到环境变量中。这一步很重要。(配置到环境变量的path全局变量里,详细过程这里就不演示了)

安装tensorflow-gpu

 

posted @ 2020-12-10 14:25  开源的Boy  阅读(129)  评论(0)    收藏  举报