今天用jupyter notebook跑代码,明明写了os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
但还是不用gpu跑给我气的
后来发现放在训练集前面就可以用gpu了
os.environ[‘CUDA_VISIBLE_DEVICES‘] = ‘0,1‘无效
最新推荐文章于 2024-09-08 18:03:17 发布
今天用jupyter notebook跑代码,明明写了os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
但还是不用gpu跑给我气的
后来发现放在训练集前面就可以用gpu了