WIN10 + GTX1660Ti配置TensorFlow GPU版本

安装准备:

         操作系统: Windows 10
  
       Python版本: 3.7.3
        安装Anaconda3-2019.03-Windows-x86_64自带python3.7.3
        下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
        参考链接:https://blog.csdn.net/junchengberry/article/details/81217645
        
        TensorFlow-GPU版本: 1.14.0
        
        Cuda版本: cuda_10.0.130_411.31_win10
        下载地址:https://developer.nvidia.com/cuda-10.0-download-archive
        
        Cudnn版本: cudnn-10.0-windows10-x64-v7.6.1.34.zip
        下载地址:https://developer.nvidia.com/rdp/cudnn-download
       
       清华镜像:
       -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

       conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
       conda config --set show_channel_urls yes

安装过程:

1.安装python
安装anaconda即可

2.安装cuda
在系统变量环境变量中path配置(已经有的不用再次配置):
在这里插入图片描述
在cmd命令窗口中输入

nvcc -V

查看cuda是否安装成功:
在这里插入图片描述
3. 安装Cudnn,解压cudnn-9.0-windows10-x64-v7.zip
将Cudnn包中所带的文件放到Cuda目录(上面的CUDA_PATH的路径)下对应的文件夹即可
Cudnn文件夹下的: bin/cudnn64_7.dll 拷贝到 Cuda文件夹下的 bin/ 目录下
Cudnn文件夹下的: include/cudnn.h 拷贝到Cuda文件夹下的 include/ 目录下
Cudnn文件夹下的: lib/x64/cudnn.lib 拷贝到Cuda文件夹下的 lib/x64/cudnn.lib目录下
(cuda默认安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0)

4.安装tensorflow-gpu

pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

5.测试

  import tensorflow as tf
    a = tf.constant(1)
    b = tf.constant(2)
    sess = tf.Session()
    print(sess.run(a + b))

输出为 3

  • 12
    点赞
  • 42
    收藏
    觉得还不错? 一键收藏
  • 18
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值