TensorFlow 1.15.8DirectML加速版
Windows 下搞TensorFlow
开发又卡又烦?试试这个:tensorflow_directml-1.15.8-cp37m-win_amd64.whl
,真挺香的。这个版本是用DirectML
加速的,专门为不装 CUDA、不想折腾 GPU 驱动的同学准备的,适合 AMD 显卡或者集显用户,能跑起来就比啥都强。
老版本的TensorFlow 1.15
虽然有点老,但架构清晰,调试友好,还兼容不少老项目。搭上DirectML
,在普通 Windows 机器上跑模型也能飞起来,基本不用折腾 CUDA 环境,安装轻松多了。
你只要用pip install tensorflow_directml-1.15.8-cp37m-win_amd64.whl
装上就能用,Python 3.7 环境下直接跑,兼容性挺稳定的。尤其在做一些小规模训练或者推理实验的时候,响应也快,体验还不错。
,如果你平时用PyTorch
多,也有类似的DirectML
方案,但这个包在 TensorFlow 上表现更成熟。可以配合keras
用来练手,或者试试自己写数据流图,效果还行。
想深入了解 TensorFlow 的数据流机制?可以看看《tensorflow-1.13.1-macosx 数据流图与自动微分机制详解》这篇文章,讲得挺细。
如果你还在找轻便的机器学习方案,又不想在 Windows 上折腾 CUDA,那这个tensorflow_directml真的值得一试。嗯,能让你的旧电脑也焕发点青春。
使用前注意:只支持 Python 3.7,其他版本会报错;而且有时候在虚拟环境里跑更稳,建议配合venv
使用。
61.35MB
文件大小:
评论区