Skip to content

Commit

Permalink
Update use-gpu.md
Browse files Browse the repository at this point in the history
  • Loading branch information
astonzhang authored Feb 22, 2019
1 parent d8d871b commit 5523733
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion chapter_deep-learning-computation/use-gpu.md
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ MXNet的计算会在数据的`context`属性所指定的设备上执行。为了
(z + 2).exp() * y
```

注意,MXNet要求计算的所有输入数据都在内存或同一块显卡的显存上。这样设计的原因是CPU和不同的GPU之间的数据交互通常比较耗时。因此,MXNet希望用户确切地指明计算的输入数据都在内存或同一块显卡的显存上。例如,如果将内存上的`NDArray`变量`x`和显存上的`NDArray`变量`y`做运算,会出现错误信息。当我们打印`NDArray`或将`NDArray`转换成NumPy格式时,如果数据不在内存,MXNet会将它先复制到内存,从而造成额外的传输开销。
注意,MXNet要求计算的所有输入数据都在内存或同一块显卡的显存上。这样设计的原因是CPU和不同的GPU之间的数据交互通常比较耗时。因此,MXNet希望用户确切地指明计算的输入数据都在内存或同一块显卡的显存上。例如,如果将内存上的`NDArray`变量`x`和显存上的`NDArray`变量`y`做运算,会出现错误信息。当我们打印`NDArray`或将`NDArray`转换成NumPy格式时,如果数据不在内存里,MXNet会将它先复制到内存,从而造成额外的传输开销。

## Gluon的GPU计算

Expand Down

0 comments on commit 5523733

Please sign in to comment.