GPU版本TensorFlow的安装(CUDA10.0+TensorFlow2.0.0+python3.6.9)

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) 环境变量的配置

GPU版本TensorFlow的安装(CUDA10.0+TensorFlow2.0.0+python3.6.9)_第1张图片
这里10.1改成10.0即可
GPU版本TensorFlow的安装(CUDA10.0+TensorFlow2.0.0+python3.6.9)_第2张图片

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了,听起来很简单,搞了我一下午……

你可能感兴趣的:(GPU版本TensorFlow的安装(CUDA10.0+TensorFlow2.0.0+python3.6.9))