安装Tensorflow-gpu踩过的坑

本文介绍了如何在CUDA8.0环境下安装Tensorflow-gpu 1.3.0,详细步骤包括环境配置、CUDA与CUDNN下载、文件替换以及可能遇到的问题与解决方法。
摘要由CSDN通过智能技术生成

1.环境配置

        Tensorflow-gpu==1.3.0   +   CUDA8.0  +  CUDNN6.0

         解释一下,现在的Tensorflow-GPU版本已经到了2.0.0a0版本,因为本人的笔记本显卡GTX-1060的显卡只能安装CUDA8.0版本,所以对应的Tensorflow-GPU版本只能选1.4.0以下的版本。

 

2.文件下载

        CUDA8.0     https://developer.nvidia.com/cuda-downloads

        CUDNN6.0  https://developer.nvidia.com/rdp/cudnn-archive

       pip install tensorflow-gpu==1.3.0 -i https://pypi.douban.com/simple

       文件下载之后,进行安装

3.替换CUDA8.0中的文件

     安装完成之后 将下载的cudnn-8.0-windows10-x64-v6.0.zip文件解压后,

      将cudnn-8.0-windows10-x64-v6.0\cuda\bin下的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

      将cudnn-8.0-windows10-x64-v6.0\cuda\include 下的文件复制到 C:\Program Files\NVIDIA GPU Compu

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值