diff --git a/docs/chapter04_DL_computation/4.6_use-gpu.md b/docs/chapter04_DL_computation/4.6_use-gpu.md index e69a92dae..f8f8061c8 100644 --- a/docs/chapter04_DL_computation/4.6_use-gpu.md +++ b/docs/chapter04_DL_computation/4.6_use-gpu.md @@ -2,7 +2,7 @@ 到目前为止,我们一直在使用CPU计算。对复杂的神经网络和大规模的数据来说,使用CPU来计算可能不够高效。在本节中,我们将介绍如何使用单块NVIDIA GPU来计算。所以需要确保已经安装好了PyTorch GPU版本。准备工作都完成后,下面就可以通过`nvidia-smi`命令来查看显卡信息了。 -```{.python .input n=1} +``` python !nvidia-smi # 对Linux/macOS用户有效 ``` 输出: