调整GPU配置
This commit is contained in:
@@ -17,4 +17,4 @@ IE_BATCH_SIZE = 4
|
|||||||
LAYOUT_ANALYSIS = False
|
LAYOUT_ANALYSIS = False
|
||||||
|
|
||||||
# 可使用的GPU
|
# 可使用的GPU
|
||||||
CUDA_VISIBLE_DEVICES = "1"
|
CUDA_VISIBLE_DEVICES = "0,1"
|
||||||
|
|||||||
Reference in New Issue
Block a user