Tensorflow-gpu==2.0.0的安装
需要下载的文件,CUDA10.0,与之对应版本的cudnn,需要更新NVIDIA显卡驱动。
一.先装好anaconda3,记得在环境里面配置好路径,一个Anaconda3,一个library lib,一个scipt。
二.直接双击安装cuda10.0,可以勾选只安装CUDA。CUDA也需要配置环境变量。
三.将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出现。**同时注意。目前tensorflow2.0版本只支持python3.5-3.7。