每周小技巧

1. 查看gpu使用情况
   nvidia-smi

使用 kill -9 进程id

1

kill -9 144953

1. 指定GPU

   1 2	import os os.environ["CUDA_VISIBLE_DEVICES"] = "2"

   命令行指定

   1	CUDA_VISIBLE_DEVICES=1 python Train.py