最近也是学了一点,现在更新一下。之前一直是在自己电脑下载好几个版本的cuda,同时还是添加环境变量特别麻烦,最近才知道在虚拟环境中也可以下载,特此更新一下。我的baseline比较老,我的版本信息是TensorFlow-gpu1.4-Python3.5-cuda8.0-cudnn6.0,不知道的伙伴可以去TensorFlow官网查询自己所需要的对应版本信息。
1.使用anconda来管理虚拟环境,建立虚拟环境,djw我是随便取的
conda create -n djw python=3.5
2.进入虚拟环境
conda activate djw
3.安装cuda和cudnn,伙伴们可以根据自己的需求去设置
conda install cudatoolkit=8.0
conda install cudnn=6.0
4.因为TensorFlow版本很老,我选择离线安装,然后conda install会出现识别路径的问题,不是很稳定,因此建议用pip安装,如果不确定要安装的版本是否在conda库里面被找到,可以使用搜索指令
conda search tensorflow-gpu
5.离线安装
cd D:\ #进入安装包目录
pip install D:\tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl
6.等待安装完成,退出虚拟环境
conda deactivate
7.检查虚拟环境是否建立成功,并根据显示位置信息,在使用pycharm选取编译器时候将虚拟环境添加即可
conda env list
8.我也是成功开始训练了