Flux应用双显卡

最近在用Flux学习深度学习方面的知识,有两张显卡,但不知道怎么将程序部署到两张显卡上并行,请问什么方法可以做到。

貌似现在只能通过跑多进程,然后数据手动分配,再手动同步参数啥的

CUDAnative.jl中暴露了device!函数,可以用于在运行时指定gpu,但是(我个人)实践中用起来没有TensorFlow那么顺畅

你可以subscribe 这个issue:
https://github.com/JuliaGPU/CuArrays.jl/issues/52

1 个赞