安装TF命令,gpu版本:
pip install tensorflow_gpu==2.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
安装好后可能会报错的几个地方:
找不到dll,可能的地方CUDA的环境变量没弄对,cudnn版本和CUDA不匹配。
cudart64_*.dll,类似100改101之类的,能自然匹配最好,不行就搜素后改下名字。运行GPU版本的时候会慢慢出现陆续的dll,全是I就对了。
另一个可能,VC_redist.x64 .exe安装一下,缺c++引导之类的。
测试GPU是否安装成功,新建个*.py, 输入