Tensorflow-gpu2.5.0+CUDA+CUDNN安装教程

Tensorflow-gpu2.5.0+CUDA+CUDNN安装教程,第1张

环境:windows10+anaconda

1.创建conda房间并激活

‘tensorflow对应python版本为3.6-3.8’

打开anaconda prompt

conda create -n tf2.5 python=3.6

激活环境

activate tf2.5

2.安装tensorflow

pip install tensorflow-gpu=2.5.0

3.安装CUDA

‘tensorflow2.5对应CUDA版本为11.2.0,下载网址如下’

CUDA Toolkit 11.2 Downloads | NVIDIA Developer下载完成后一路默认安装(只用CUDA进行gpu加速,所以没有vs也无所谓)

4.cudnn

cuDNN Archive | NVIDIA Developer‘’

cudnn下载地址 

选择下图版本

选择windows x86版本

 

下载完成,解压压缩包,将解压后的三个文件夹替换为CUDA安装路径中的文件夹,默认安装路径为:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/942639.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-18
下一篇2022-05-18

发表评论

登录后才能评论

评论列表(0条)

    保存