python - 没有 GPU 使用 torch-PinSage

Question

我正在使用Python 3.6、CUDA 10.1、torch 1.7.1+cu101、dgl-cu101版本0.6.1；但是，当我运行程序时（这是此处提供的PinSage的实现： https ://github.com/dmlc/dgl/tree/master/examples/pytorch/pinsage ），以及任务管理器和 GPU 中的 GPU 使用情况- Z为零！我还运行了以下命令：

torch.cuda.is_available()
torch.cuda.current_device()

他们正在返回指示 GPU 使用情况的结果。任务管理器和GPU-Z的问题怎么办？（当然，运行时间也很短，因为我将它与 Google Colab GPU 运行时进行了比较。）

我有 NVIDIA GEFORCE 940MX。