tensorflow-gpu的安装

标题tensorflow-gpu安装

Win10anaconda3/python.exe(3.7版),使用的是anaconda里的python,不需要再自己安装python。如果有GPU可以安装tensorflow-gpu,没有GPU就安装tensorflow
下是讲tensorflow-gpu的安装2020-3-9):
最重要的是CUDA,cuDNN,tensorflow-gpu版本一定匹配,这个必须要搞清楚不然后面还得卸载。https://www.tensorflow.org/install/source_windows鼠标滑到底,有对应的版本。
我的版本是CUDA(10.0)+cuDNN(7.6.5)+tensorflow-gpu(2.0)
CUDA的安装https://developer.nvidia.com/cuda-toolkit-archive:这个是英伟达CUDA的历史版本,找到需要的版本号,不要直接下载最新版的安装包。下载安装包后一切默认安装即可。安装完成后可通过cmd里面输入nvcc -V查看CUDA版本号。
cuDNN的安装:英伟达官网下载: https://developer.nvidia.com/rdp/cudnn-download 下载对应的版本,需要注册,可以用微信号或者qq注册。
下载完成解压缩之后,把bin,include,lib/x64里的文件分别放入CUDA对应的bin,include,lib/x64里面。CUDA的默认路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
利用清华源安装tensorflow-gpu:

pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

完美!

发布了2 篇原创文章 · 获赞 0 · 访问量 7

猜你喜欢

转载自blog.csdn.net/weixin_44546620/article/details/104752596