useful code

在这里记录下有用的代码，也是提醒自己在跑实验的时候不要只关注于代码正确了，实验跑通了，深入理解代码细节！持续更新中📚…

设置可见GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = gpu_ids
gpu_ids = '0'

• 指定GPU：通过设置 CUDA_VISIBLE_DEVICES 环境变量，可以指定哪些GPU可供程序使用。例如，如果有4个GPU（编号为 0, 1, 2, 3），但只希望程序使用GPU 2和GPU 3，你可以设置gpu_ids = "2,3"，这样程序运行时只会看到这两个GPU。
• 屏蔽GPU：如果不希望程序使用任何GPU，可以将gpu_ids设置为空字符串，即gpu_ids = ""。这会使得程序无法看到任何GPU，从而只使用CPU。

attention: 这行代码实际上只是告诉程序哪些GPU是可见的或可用的，但它并不保证程序一定会使用这些GPU中的所有GPU。