由于本地安装了cuda 10.0, 但是现在需要在Anaconda中安装不同的python虚拟环境来安装tensorflow-gpu、对应的cudatoolkit、对应的cudnn来加速,下面是具体的演示流程
2. 安装
我这里以安装tensorflow-gpu==1.9.0为例,首先进入python的虚拟环境,source activate py37_tf
通过"conda search tensorflow-gpu", 可以看到有满足安装版本要求的tensorflow-gpu
然后使用清华源安装: pip install tensorflow-gpu==1.9.0 -i Simple Index
安装成功以后去网络上搜索"tensorflow和GPU的对应关系"
可以看到需要对应安装cudatoolkit==9.0, 我们在通过命令: “conda search cudatoolkit"来看一下是否有此cudato