如果在命令行中 CUDA_VISIVIBLE_DEVICES=8 python main,py但是程序中还是例如默认用了0卡,那就是命令行CUDA指定失效,原因是
①作者可能是将cuda的指定写成了一个argparse,查看一下有没有这一项, 有的话注释掉
②作者可能通过os.environ["CUDA_VISIBLE_DEVICES"] 等设置
如果暂时查不出问题,用这句
os.environ["CUDA_VISIBLE_DEVICES"] = "4,5,6,7"
在这种情况下,程序会把4卡认为0卡,所以它如果再报GPU 0,指的就是4卡
命令行CUDA指定失败
最新推荐文章于 2023-03-12 11:01:55 发布