admin 管理员组

文章数量: 887016

背景简介

作为刚入门深度学习的小白,准备兴高采烈的开始用GPU跑深度学习算法.已经安装好了tensorflow的CPU版本,都是想要加速,搭建tensorflow-gpu的版本。

我的环境是python==3.7, tensorflow==1.15,这两个的安装可以去参考其他教程,为什么在2021年还选择1.15的版本,因为我要用的代码还是tensorflow 版本1的,tensorflow 2.0的和原来的不一样,容易出现其他问题,为了早日上手深度学习,就不去弄2.0了。这里主要是讲解如何安装GPU加速CUDA和cuDnn。

tensorflow-gpu-1.15.0,tensorflow-gpu安装直接在命令行窗口输入:pip install tensorflow-gpu==1.15,注意要加上版本号,不然直接会安装最新版的。会和原来安装的版本不匹配。

显卡支持:

明确的一点是,我已经假定大家的显卡都能支持cuda了,基本现在的显卡都支持cuda了,除非你的机子非常老了,如果你仍然不放心,就去这个网址去查吧https://developer.nvidia/cuda-gpus;如果没找到自己的显卡型号,那就可能是不支持了。

cuda安装:

下载地址:https://developer.nvidia/cuda-toolkit-archive

这里要特别注意,不知道为什么一开始我一直下

本文标签: 教程 TensorFlow GPU