环境: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