简单易学:Tensorflow CUDA 版本安装

162 阅读1分钟
各个平台通用账户:一个有趣的灵魂W
看了那么多Tensorflow(GPU)安装教程,一个字,迷糊!

来一期简版教学吧:
1、想用GPU,就得确定你的电脑支不支持CUDA:
这里各位可以直接百度你GPU的型号,不展开了。
2、打开英伟达官网,选择对应版本:
https://tensorflow.google.cn/install/source_windows?hl=zh_cn
(包括python版本、编译器、MSVC、cuDNN、CUDA)

2.1
看好版本后,就照着下就完了~ Python(3.7)我使用Anaconda,具体参照以下:
python都拿你没办法,历史版本Anaconda下载​mp.weixin.qq.com
https://mp.weixin.qq.com/s?__biz=Mzg5NDIxNjM1Ng==&mid=2247484612&idx=1&sn=5d97dc0f038afbffb9d3bb6e97bf0036&chksm=c023b50ff7543c19a2273e84520d16bcebc0a42d23b20fe00ad5025a144f768a44abd8185fe8&token=1902223798&lang=zh_CN#rd
注:为什么选择3.7?
你敢信?新版的Python居然因为这个原因被束缚住了。​mp.weixin.qq.com
https://mp.weixin.qq.com/s?__biz=Mzg5NDIxNjM1Ng==&mid=2247484634&idx=1&sn=e7ed31ed68dd755da64002a1440a0140&chksm=c023b511f7543c07a548694d62fefd26d9eff513cb9cf66844231f5b87401c3cf70465465fb0&token=1902223798&lang=zh_CN#rd
2.2
MSVC--该处我选择2019版本
https://visualstudio.microsoft.com/zh-hans/vs/

接着安装,选择单个组件:

安装就完了。
2.3 cuDNN,
选择对应7.4.(实际上cuDNN中7.4没有对应的CUDA10.1,所以我们选择了7.5)。cuDNN需要先注册:
https://developer.nvidia.com/rdp/cudnn-archive
注册后选择:

下载好后,解压,将bin添加到环境变量中。(环境变量如何添加,可以百度一下)
https://baijiahao.baidu.com/s?id=1652500747054512497&wfr=spider&for=pc
路径:

path

2.4 CUDA
CUDA-下载CUDA10.1,如果你电脑CUDA版本不匹配,可以先卸载CUDA或者英伟达软件,继而安装需要的版本。下载地址:
https://developer.nvidia.com/cuda-toolkit-archive

3、安装Anaconda
你可以直接从官网下,也可以选择下载历史版本:
https://mp.weixin.qq.com/s?__biz=Mzg5NDIxNjM1Ng==&mid=2247484612&idx=1&sn=5d97dc0f038afbffb9d3bb6e97bf0036&chksm=c023b50ff7543c19a2273e84520d16bcebc0a42d23b20fe00ad5025a144f768a44abd8185fe8&token=1902223798&lang=zh_CN#rd
4、安装Tensorflow(GPU)
建议官网下载。
https://tensorflow.google.cn/install/pip?hl=zh_cn

(下载器下载就得了)接着安装CUDA。
5、最后pip
给小白提个醒,下载好的whl文件安置在D盘根目录下,然后写入
pip install tensorflow_gpu-2.3.0-cp37-cp37m-win_amd64.whl

接着测试,显示是True就是装好了。。。(收工)

更多细节,参照:
简单易学:Tensorflow CUDA 版本安装​
https://mp.weixin.qq.com/s?__biz=Mzg5NDIxNjM1Ng==&mid=2247484670&idx=1&sn=477b79f8101d5c5d2d694055e6d85ab7&chksm=c023b535f7543c23d0ea983d4117a70f45f398c5e73c8f5869df7e47723a0deaf83444634d9e&token=1902223798&lang=zh_CN#rd