tnfh.net
当前位置:首页 >> tEnsorFlow 1.0.1 指出CuDnn8么 >>

tEnsorFlow 1.0.1 指出CuDnn8么

安装GPU版本的tensorflow就可以了pipinstall--upgrade还需要安装GPU加速包,下载CUDA和cuDNN的安装包了,注意版本号分别是CUDA8.0和cuDNN5.1。 如果都安装成功,运行程序时会自动使用GPU计算

需要看你的PC配置是否够,TF的GPU模式只支持N卡,然后计算能力高于3.0,以下说下安装步骤: 1、首先要注册NVIDIA developer的帐号,分别下载CUDA和cuDNN。 2、确认准备gcc版本,安装依赖库sudo apt-get install freegl。 3、安装CUDA。 4、解压c...

网站首页 | 网站地图
All rights reserved Powered by www.tnfh.net
copyright ©right 2010-2021。
内容来自网络,如有侵犯请联系客服。zhit325@qq.com