Tensorflow-gpu==2.0.0的安装

需要下载的文件,CUDA10.0,与之对应版本的cudnn,需要更新NVIDIA显卡驱动。

一.先装好anaconda3,记得在环境里面配置好路径,一个Anaconda3,一个library lib,一个scipt。

二.直接双击安装cuda10.0,可以勾选只安装CUDA。CUDA也需要配置环境变量。Tensorflow-gpu2.0.0版本Windows安装简要流程

三.将cudnn解压,根据CUDA的安装目录,bin对应bin,lib对应lib,include对应include,将cudnn里面内容分别复制进去。

四.进入anaconda prompt,pip install tensorflow-gpu==2.0.0 -i http://pypi.douban.com/simple,豆瓣源比清华源快,且我在使用清华源时会出现一堆红字bug,豆瓣源不会。

五.ipython,import tensorflow as tf,tf __version__看看是否成功。

注意:tensorflow版本与CUDA和cudnn版本皆有对应,现阶段2.0.0使用非10.0版本的CUDA会有意想不到的bug出现。

相关文章: