一、简介:
GPU在深度学习中是至关重要的。GPU非常适合并行计算,而神经网络是高度并行的(embarrassingly parallel)。在并行计算中,高度并行任务是指将整个任务分割成一组较小的任务以并行计算的任务。高度并行任务是那些很容易看到一组小任务彼此独立的任务。
二、环境介绍:
操作系统:windows 10
python版本:python 3.6.5
其他:CUDA 10.1
cuDNN 7.6.5
tensorflow 2.3.1
keras 2.4.3
英伟达官方驱动下载:http://www.nvidia.cnhttp://www.nvidia.cn
注意:先更新自己显卡驱动到最新或最稳定版本
三、安装:
python的安装这里就不阐述了。官网下载后按提示下一步即可。
tensorflow 和 keras 都可以直接用 pip安装:
pip ins