当前位置: 首页 > 网络应用技术

Win10 TensorFlow-GPU环境详细的步骤教程

时间:2023-03-08 16:36:58 网络应用技术

  昨天,我努力匹配GPU环境,并将其记录为将来。

  我的GPU的目的是用张量的GPU加速

  第一个是安装cuda1的nvidia的驱动程序,首先安装cuda10.1 cuda下载地址

  选择适合您下载时的版本

  上面找到的最新版本的CUDA。我们想找到的是版本10.1的数据

  下载下载,让我们安装

  只需直接安装

  下一步将选择安装路径。我首先是自己设置路径,因为我不想将其放在C驱动器上,但是在安装后,我没有在设置的路径中找到安装CUDA。因此,我安装了它。第二次,我没有卸载第一个安装,因为我没有找到卸载它的方法。直接启动第二个安装,或者自己选择设置的路径,或者安装不成功。它不舒服,因此,直接选择了第三个安装以选择其默认路径。可以完全接受多个G中的一个,并且将其放入C驱动器中不是一个问题。这次安装成功。

  因此,个人建议,如果C驱动器可以接受多个G,则最好将其直接安装在默认路径上以防止旧问题。

  安装完成后,需要下一步来配置环境变量。

  如果安装成功,系统变量将多两个作为额外的

  在安装9.0时,使用的图片是保留的图片。

  您还需要自己添加一些系统变量

  添加后,应该像以下

  在下一步中将内容添加到系统变量。double-Click查找系统变量路径

  添加以下内容。

  添加后,安装了CUDA。我们可以测试安装是否成功。

  打开CMD,CD至C:Program filesnvidia gpu计算ToolkitCudav9.0ExtrasDemo_suite(这是默认路径)

  单独执行命令:

  在正常情况下返回

  CUDA10.1的安装成功。

  安装Cudnn.Go访问官方网站以下载使用CUDA10.1的Cudnn,请务必注意匹配的10.1.我们选择Cudnn7.6.5,这将需要注册NVIDIA帐户。

  下载后,解开拉链,将.h .lib和.dll文件复制到cuda的lib/x64,cuda的bin文件夹。实际上,它是相应的文件夹。

  CUDA10.1及其相应的Cudnn在这里安装。

  安装TensorFlow-GPU

  测试是否成功

  输出真实,证明安装成功