#Linux如何查看CUP及其使用情况
1.先下载gpustat: pip install gpustat
2.watch -n 2 --color gpustat --c 即可查看这里可以看到4是可用的,然后我们再用DEVICE = torch.device(“cuda:4”)设置需要使用的GPU
2024-01-24 20:17:40
#Linux如何查看CUP及其使用情况
1.先下载gpustat: pip install gpustat
2.watch -n 2 --color gpustat --c 即可查看这里可以看到4是可用的,然后我们再用DEVICE = torch.device(“cuda:4”)设置需要使用的GPU