Curvature-Learning-Framework icon indicating copy to clipboard operation
Curvature-Learning-Framework copied to clipboard

gpu训练

Open yifenzhong1920 opened this issue 3 years ago • 0 comments

使用gpu训练的时候,显存有使用,但是gpu没有使用?使用的是CUDA_VISIBLE_DEVICES=0 python examples/hgcn/train.py 运行的hgcn. example

yifenzhong1920 avatar Dec 22 '21 08:12 yifenzhong1920