失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > 在CUDA10的环境下安装支持gpu的tensorflow版本(亲测有效)

在CUDA10的环境下安装支持gpu的tensorflow版本(亲测有效)

时间:2019-12-01 10:56:55

相关推荐

在CUDA10的环境下安装支持gpu的tensorflow版本(亲测有效)

由于英伟达官方宣称CUDA10有着巨大的性能提升,我便迫不及待的想尝试用它来训练深度学习模型,然而现在Tensorflow的最新官方GPU版本1.12并不支持在CUDA10下的模型训练,因此便翻墙到国外论坛找材料,最终找到了可以完美适配CUDA10的tensorflow方案。具体的环境如下:

安装步骤:

首先将相关的轮子依赖下载到本地系统中,(/download/weixin_38538282/10816219或者翻墙下载:https://1drv.ms/u/s!AspzPyzQ0HQR7VNCicPlnCV2abKJ)我选择的是Pycharm工程venv目录下的Scripts,然后再进入到该目录下,输入pip3 install tensorflow-1.12.0rc0-cp36-cp36m-win_amd64.whl 命令,即可成功安装CUDA10下支持的tensorflowGPU版本。

出现下图即说明已经成功使用GPU训练该模型:

最后贴一张训练效果图:

您的赞赏是对我继续创作的支持和鼓励:)

如果觉得《在CUDA10的环境下安装支持gpu的tensorflow版本(亲测有效)》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。