服务器显卡选择方法

想让服务器上的程序跑在指定GPU上，核心答案就是通过环境变量CUDA_VISIBLE_DEVICES和框架自带的设备参数（如PyTorch的device）来精确控制，设置后程序只会看到你指定的那一块卡，怎么让程序在指定gpu上跑？先看懂CUDA_VISIBLE_DEVICES很多人在单卡环境下写代码，一上多卡服……