tensorflow_gpu版本安装
这是一个非常简洁的安装指南,务必严格按照上述要求检查安装。
https://zhuanlan.zhihu.com/p/183981491
利用anaconda安装 tensorflow_gpu 2.3.0,一开始因为版本一致问题,一直很惧怕安装,最终自己想了最简单的方法,和上面链接方法基本一致,链接介绍了原理!
1.创建Anaconda虚拟环境,同时指定python版本。
打开anaconda prompt 命令行,执行以下命令:
conda create -n tfgpu2.3.0 python=3.7
2.检查nvidia驱动版本,NVIDIA驱动程序需 418.x 或更高版本。
可在命令行中执行查看驱动版本: nvidia-smi
3.安装依赖库
如上图,可知CUDA版本是11.1。 GPU版本有两个依赖库cuda和cudnn,对于 tensorflow2.3来讲,查阅资料知道:
CUDA的版本需要是 10.1
cudnn版本号需要不小于 7.6
如果直接执行conda install cudatoolkit cudnn,anaconda会自动匹配版本,CUDA的版本是 10.2,我不知道可不可以,有的文章说会产生冲突,我是按照指定版本的方式安装!
3.执行下面命令,cudnn自动匹配的版本应该是7.6.5:
conda install cudatoolkit=10.1 cudnn
4.最后执行tensorflow-gpu安装:
conda install tensorflow-gpu=2.3.0
5.验证安装成功与否,按照图片的命令执行即可!