import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1" #指定第一块gpu
python指定GPU
最新推荐文章于 2024-10-04 08:15:00 发布
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1" #指定第一块gpu