随着网络越来约复杂,训练难度越来越大,有条件的可以采用GPU进行学习。本文介绍如何在GPU环境下使用TensorFlow.NET。

TensorFlow.NET使用GPU非常的简单,代码不用做任何修改,更换一个依赖库即可,程序是否能运行成功主要看环境是否安装正确,这篇文章重点介绍的也就是环境的安装了。

CUDA和cuDNN的安装都比较容易,重点是要装对版本。

1、确认安装版本

首先电脑得有一块NVIDIA的显卡!

在桌面右键选择NVIDIA控制面板,在程序左下角点击 :系统信息。

12-31 11:26