显卡指定问题 

指定运行环境


import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

　　

posted @ 2022-02-09 17:42  hifalee  阅读(29)  评论(0)  编辑  收藏  举报