关于Tensorflow和Pytorch双框架使用GPU加速

关于Tensorflow和Pytorch怎么使用GPU加速

关于tensorflow和pytorch怎么使用gpu加速，作为一名深度学习的爱好者，总要使用gpu加速程序计算，我在使用这些框架时遇到的一些坑，给大家推荐一下使用gpu加速的方法，帮助大家避坑。下面开始。第一个是tensorflow的gpu加速，第二是pytorch的gpu的加速

---

前言提示

提示：一定要看自己的显卡的驱动程序，看清楚自己的cuda能够安装的版本，通过控制面板上看到的驱动程序（这个程序可以更新的，如果不够高，可以通过更新来升高版本）,要安装的cuda版本是在这个驱动程序之下的。

---

一、tensorflow框架下的gpu加速

1、推荐使用anaconda+pycharm进行操作
2、打开anaconda，新建自己的环境库（这个步骤大家应该都会）
3、打开pycharm新建自己的工程，使用自己新建的anaconda环境库作为编译器。
4、之后就进入了新建的工程了，打开pycharm的终端窗口。如下图所示。

5、根据自己的tensorflow版本安装合适的cudatoolkit和cudn