坑
指定GPU
先用nvidia-smi查看gpu负载状态,然后跑torch的时候,指定可见的gpu ID就行
在运行python blablabla前面加一句变成CUDA_VISIBLE_DEVICES=0,1,2 python blablabla,意义是可见的gpu有0,1,2号,而且torch会按照0,1,2这个顺序来使用gpu
链接 pytorch中tenso...
坑
指定GPU
先用nvidia-smi查看gpu负载状态,然后跑torch的时候,指定可见的gpu ID就行
在运行python blablabla前面加一句变成CUDA_VISIBLE_DEVICES=0,1,2 py...