PyTorch使用技巧

问题现象：明明有两张卡，为何只能看到一张？ 很多人在使用配备多块GPU的服务器时，都遇到过这样一个奇怪的现象：通过nvidia-smi命令能看到系统中有两张显卡，但在PyTorch等深度学习框架中运行torch.cuda.device_count时，返回的结果却是1，意味着程序只能识别到一张GPU卡。 这种情况在深度学习开发和模型训练中尤为常见。当你尝试指定…