2021年10月27日
torch设置GPU
摘要： 设置torch可见哪几个GPU： import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" # 例子>> "0"：只有0号GPU可见；"1,3"：只有1号和3号GPU可见；""：全部不可见，此时 torch.cuda.is_available() 输出 阅读全文
posted @ 2021-10-27 09:39 乌蝇哥 阅读(2409) 评论(0) 推荐(0) 编辑