装了三天的tensorflow-gpu,看了无数的博客,试了无数次方法,踩了无数的坑,今天终于给他喵的整好了。写个博客给大家填个坑。
首先查一下显卡配置:
在命令行中输入nvidia-smi
这里可能需要配置环境变量C:\Program Files\NVIDIA Corporation\NVSMI
然后去官网找到对应关系:
由于我的是399.24所以我装的cuda版本应该是9.2.148
再去找对应tensorflow的版本win10下的
看以看出我可以装1.6到1.13,cudnn的版本为7.
OK,准备工作做完了。现在去下载anacondawin10 64bit按照步骤一步一步来就能安好,这挺简单的。但是要注意要选上Add … PATH这样就不用添加环境变量了。
在装完后可以改一下conda的下载源。这个可以参考我以前写的博客换清华源
现在导包:
命令行中创建虚拟环境conda create -n tensorflow-gpu python=3.6
这里建议装3.6。
然后激活环境activate tensorflow-gpu
就是进入刚才间的文件夹(tensorflow-gpu)
导入tensorflow包conda install tensorflow-gpu=1.13
版本根据自己的选择
在这里个地方可能会出错
可以看出这两个并不适合我的版本,别急先导里面。
下载完成后conda install cudatoolkit=9.2
conda install cudnn=7
然后去nivdia官网下载cuda和cudnncudatoolkit
cudnn
CUDA位置默认就好方便下面的操作,cudnn下载需要注册。
下载完成后将cudnn文件解压分别将文件夹下面的文件copy到CUDA对应的文件夹下。
到这个地方应该完成一大半了。这样就可以在命令行中运行了。(要先激活环境,也就是进入tensorflow-gpu文件夹)
想要在pycharm中运行的话还需要在pycharm中进行配置
首先需要 project interpreter 这个很简单就是找到这个文件夹下的python.exe
然后就是将这个弄好
配置 Edit Configurations
现在就都弄好了(我的是这样)。