search

yao8839836 / kg-bert

KG-BERT: BERT for Knowledge Graph Completion
Apache License 2.0
679 stars 141 forks source link

GPU的使用 #23

Open nxtabb opened 3 years ago

nxtabb commented 3 years ago

您好，请问我下载了您的代码，并尝试进行复现，但是在训练，验证，测试的过程中，gpu的使用只有显存被占用了，而并没有使用GPU进行运算，您知道是怎么回事吗？我需要在代码中添加什么吗？

yao8839836 commented 3 years ago

@nxtabb

您好。可能是这一句没有对应上指定的显卡，要根据实际情况修改

os.environ['CUDA_VISIBLE_DEVICES']= '3'

nxtabb commented 3 years ago

好的 解决了 谢谢您