今天用jupyter notebook跑代码,明明写了os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
但还是不用gpu跑给我气的
后来发现放在训练集前面就可以用gpu了
版权声明:本文为weixin_42844368原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
今天用jupyter notebook跑代码,明明写了os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
但还是不用gpu跑给我气的
后来发现放在训练集前面就可以用gpu了