前几天重装了系统,把以前的TensorFlow环境也要重新搭建了,这下还费了些功夫,真不知道我以前是怎么稀里糊涂把环境搭建好的,这里记录下,希望能够大家些帮助。
1. 安装Anaconda
我这里用的是Anaconda3-5.1.0-Windows-x86_64.exe版本。
2.安装pycharm
pycharm-community-2018.2.4.exe
3.安装cuda
我就是这步搞了太久时间,因为python版本,tensorflow-gpu,cuda,cudaDnn都有对应关系。我python3.6.4,tensorflow-gpu1.11.0,cuda版本应该是9.0版本,注意这里还有几个补丁包需要安装,我就是一开始这几个补丁包没装,导致折腾了很久,
cuda_9.0.176_win10.exe
补丁包有:
cuda_9.0.176.1_windows.exe
cuda_9.0.176.2_windows.exe
cuda_9.0.176.3_windows.exe
cuda_9.0.176.4_windows.exe
4.下载cudaDnn
我下载的是cudnn-9.0-windows10-x64-v7.1,用里面文件替换NVIDIA GPU Computing Toolkit\CUDA\v9.0里的目录即可。
5.创建anaconda虚拟环境
安装必要的依赖,tensorflow-gpu 1.11.0 等