win10下tensorflow、pytorch相关环境配置

win10下安装tensorflow-gpu2.0.0

显卡GTX 1660Ti

介于Nvida官网未给出GTX 1660Ti的具体算力,可以参考cuda的维基百科https://en.wikipedia.org/wiki/CUDA

  1. 下载CUDA版本: cuda_10.0.130_411.31_win10

    链接:https://developer.nvidia.com/cuda-10.0-download-archive

  2. 下载cuda10.0对应得cudnn版本: cuDNN v7.6.5

    链接:https://developer.nvidia.com/rdp/cudnn-download,直接下载较慢,建议右键复制链接采用迅雷下载。

  3. 安装CUDA,此部分操作详细见https://blog.csdn.net/abc13526222160/article/details/88780278

    在cmd中输入 nvcc -V查看cuda是否安装成功。

  4. 配置cudnn,此部分操作按https://blog.csdn.net/weixin_43318717/article/details/94433790

    Cudnn文件夹下的: bin/cudnn64_7.dll 拷贝 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
    Cudnn文件夹下的: include/cudnn.h 拷贝 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
    Cudnn文件夹下的: lib/x64/cudnn.lib 拷贝 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
  5. 安装tensorflow-gpu

    activate python36 #激活python36这个环境
    pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #GPU版本
    pip install tensorflow==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #CPU版本
    #验证
    import tensorflow as tf
    print(tf.test.is_gpu_available())
    #显示Ture表明安装成功
    

conda虚拟环境操作


                
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值