pytorch学习日记(三)——之Data Parallelism

PyTorch中使用了张量类型,而不用numpy的array,就是为了能够在GPU上运行代码,那咱们怎么样才能使用GPUs来加速运行呢。其实很是简单,几条语句就能够完成了。python 1. 基本语句 模型放到一个GPU上运行 model.gpu() 将张量放到GPU上 mytensor = my_tensor.gpu() 注: 仅仅调用my_tensor.gpu()函数并不会将张量复制到GPU上
相关文章
相关标签/搜索