||
主要介绍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
示例参考教程(具体硬件要求):
RTX2080Ti+Win10+Cuda10.1+Cudnn7.6+tensorflow-gpu2
特别注意:用conda安装仅限于当前环境使用,而且会自动安装相应的cudnn和cuda支持,不需要手动安装,因此建议使用conda安装相应的TensorFlow版本。
Archiver|手机版|科学网 ( 京ICP备07017567号-12 )
GMT+8, 2024-12-27 00:05
Powered by ScienceNet.cn
Copyright © 2007- 中国科学报社