Win10下GPU版本tensorflow安装
由于tensorflow默认安装都是CPU版本,之前一直安装gpu版本出现很多问题,这里终于找到一种安装方法,以及自己电脑对应的版本。供大家参考。其中找到的比较优质的博文参考如下
(1)[精简]2023年最新windows安装GPU版本tensorflow
(2)CUDA12.2版本下tensorflow-gpu安装教程
(3)机器/深度学习笔记:tensorflow(gpu版本)在win10系统上的安装方法
安装CUDA
在NVIDIA官网上安装cuda toolkit,这里安装的是最新的12.3版本的。
安装过程直接按照推荐设置,安装精简版本,未进行修改。安装完成后在下面路径下出现12.3的文件夹
安装CUDNN
也是NVIDIA官网上安装,然后解压到CUDA所在的文件夹下。在我的电脑-高级电脑设置-环境变量-PATH编辑,增加以下两个路径
在anconda环境中安装tensorflow-gpu
即使cuda和cudnn安装正确,稍不注意,还是可能由于版本问题而导致gpu版本的tensorflow无法使用。按照下面流程,再加上不断尝试,最终总结出了两个版本的内容供大家选择。当然,大家也可以根据电脑配置自行选择,选择依据主要参考官网上的对应版本。
查看cuda/cudnn版本
conda search cuda
conda search cudnn
出现的内容如下
不同版本Python安装
Python3.8版本
conda create -n tensorflowgpu python=3.8.18
conda search cuda
conda search cudnn
conda install cudatoolkit=10.1.243
conda install cudnn=7.6.5
pip install tensorflow-gpu==2.7.0
pip install matplotlib==3.3.4
中间还遇到一些错误,上面的cuda包似乎不太对,,,可能我的电脑有高版本的CUDA和cudnn,反正最终的包版本如下所示
Python3.6版本
在版本二中,只能安装2.2.0版本的tensorflow
安装的命令如下
conda create -n tensorflownew python=3.6.13
conda search cuda
conda search cudnn
conda install cudatoolkit=10.1.243
conda install cudnn=7.6.5
pip install tensorflow-gpu==2.2.0
pip install matplotlib==3.3.4
安装的包版本如下,把matplotlib特意标注出来,因为之前遇到过高版本的matplotlib对应的numpy版本与tensorflow版本要求不一致。这里采用的是低版本的3.3.4版本基本上都可以。