Open nxtabb opened 3 years ago
您好,请问我下载了您的代码,并尝试进行复现,但是在训练,验证,测试的过程中,gpu的使用只有显存被占用了,而并没有使用GPU进行运算,您知道是怎么回事吗?我需要在代码中添加什么吗?
@nxtabb
您好。可能是这一句没有对应上指定的显卡,要根据实际情况修改
os.environ['CUDA_VISIBLE_DEVICES']= '3'
好的 解决了 谢谢您
您好,请问我下载了您的代码,并尝试进行复现,但是在训练,验证,测试的过程中,gpu的使用只有显存被占用了,而并没有使用GPU进行运算,您知道是怎么回事吗?我需要在代码中添加什么吗?