GPU版本TensorFlow的安装
1. 总述
真是搞死我了,在Windows下装这些第三方包真的是麻烦死了,Conda又坏掉了没办法直接conda install,搞得要用pip,出现一堆问题。最主要的问题还是CUDA的版本和TensorFlow的版本不匹配,最开始用的是CUDA10.1和tf2.0.0,结果没有办法匹配,tf去调的是CUDA的10.0版本的dll,报错cudart64_100.dll not found……然后暴力把dll的名字从101改成100,发现可以import tensorflow但是却没办法调包里的库……一直报错Internal: invalid device function,最后觉得还是得装一下CUDA10.0,结果又搞了半天,最后终于弄好了。
2. CUDA的安装
a) 官网下安装包,记得右键复制链接地址,然后打开迅雷用迅雷下,速度快了10倍以上……
b) Cudnn的下载,直接下就好,下载完就把相应的文件夹移到CUDA的相应文件夹的位置就好了
c) 环境变量的配置
这里10.1改成10.0即可
3. TensorFlow的安装
直接在anaconda开一个环境,然后用清华的源pip就可以,记得下载的是GPU版本的TensorFlow: pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu 2019.10.02默认的是2.0.0版本的TensorFlow 装完了就OK了,听起来很简单,搞了我一下午……