NJU1healer的个人博客分享 http://blog.sciencenet.cn/u/NJU1healer

博文

Windows10系统CUDA与TensorFlow的配置

已有 462 次阅读 2021-10-19 14:21 |个人分类:机器学习|系统分类:科研笔记

主要介绍CUDA与TensorFlow的环境搭建配置的大体思路,具体细节根据自己情况调整!

(1)Windows查看CUDA版本号

详情具体见:Windows系统查看CUDA版本号

原因:在配置Tensorflow时要求与CUDA版本相对应,本文配置主要是CUDA 11. 因此,弄清CUDA版本号很重要。

(2)win10环境下CUDA和CUDNN的安装

具体参考:win10下CUDA和CUDNN的安装(超详细)!亲测有效!

                 Win10+Tensorflow2 + CUDA11 GPU环境搭建

另外,tensorflow-gpu版本国内镜像源tensorflow-gpu 清华镜像源

(3)TensorFlow与CUDA对应版本号参考,选择安装对应的TensorFlow版本(GPU).

网址:https://tensorflow.google.cn/install/source_windows

image.png

image.png

示例参考教程(具体硬件要求):

RTX2080Ti+Win10+Cuda10.1+Cudnn7.6+tensorflow-gpu2

特别注意:用conda安装仅限于当前环境使用,而且会自动安装相应的cudnn和cuda支持,不需要手动安装,因此建议使用conda安装相应的TensorFlow版本




http://wap.sciencenet.cn/blog-3428464-1308544.html

上一篇:[转载]ERROR 4: Unable to open EPSG support file gcs.csv 问题快速解决方案
下一篇:更好理解F1分数和Kappa系数

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...
扫一扫,分享此博文

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2021-11-30 15:00

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部