首页 > 其他 > 详细

简单易学:Tensorflow CUDA 版本安装

时间:2020-10-28 19:49:02      阅读:38      评论:0      收藏:0      [点我收藏+]

 

各个平台通用账户:一个有趣的灵魂W
 
看了那么多Tensorflow(GPU)安装教程,一个字,迷糊!
 
技术分享图片
来一期简版教学吧:
 
1、想用GPU,就得确定你的电脑支不支持CUDA:
这里各位可以直接百度你GPU的型号,不展开了。
 
2、打开英伟达官网,选择对应版本:
  •  
(包括python版本、编译器、MSVC、cuDNN、CUDA)
技术分享图片
2.1
看好版本后,就照着下就完了~ Python(3.7)我使用Anaconda,具体参照以下:
python都拿你没办法,历史版本Anaconda下载?mp.weixin.qq.com
 
注:为什么选择3.7?
 
你敢信?新版的Python居然因为这个原因被束缚住了。
技术分享图片
2.2
MSVC--该处我选择2019版本
 
技术分享图片
接着安装,选择单个组件:
技术分享图片
 
 
安装就完了。
 
2.3 cuDNN,
选择对应7.4.(实际上cuDNN中7.4没有对应的CUDA10.1,所以我们选择了7.5)。cuDNN需要先注册:
 
  •  
注册后选择:
 
技术分享图片
 
下载好后,解压,将bin添加到环境变量中。(环境变量如何添加,可以百度一下)
 
  •  
路径:
 
技术分享图片
path
技术分享图片
 
 
 
2.4 CUDA
CUDA-下载CUDA10.1,如果你电脑CUDA版本不匹配,可以先卸载CUDA或者英伟达软件,继而安装需要的版本。下载地址:
 
  •  
 
技术分享图片
技术分享图片

3、安装Anaconda
你可以直接从官网下,也可以选择下载历史版本:
历史版本Anaconda下载F君的小尾巴,公众号:一个有趣的灵魂W
python都拿你没办法,历史版本Anaconda下载?
m技术分享图片
 
4、安装Tensorflow(GPU)
建议官网下载。
 
  •  
技术分享图片
(下载器下载就得了)接着安装CUDA。
 
5、最后pip
给小白提个醒,下载好的whl文件安置在D盘根目录下,然后写入
 
  •  
pip install tensorflow_gpu-2.3.0-cp37-cp37m-win_amd64.whl
 
技术分享图片
接着测试,显示是True就是装好了。。。(收工)
 
技术分享图片
技术分享图片
更多细节,参照:
简单易学:Tensorflow CUDA 版本安装
 
往期
技术分享图片
 
技术分享图片
 

 

简单易学:Tensorflow CUDA 版本安装

原文:https://www.cnblogs.com/bourgeoiss/p/13891613.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!