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使用。

whl 文件大小:61.35MB