如何解决服务器显卡只在第一块运行的问题？

最近做实验，发现尽管使用了
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘2’
还是只用了第0块卡的问题
这里找到了答案，我们只需要再最前面就写这两行代码。
因为import torch之类的其他包或者依赖包可能回重置或者设定好而导致之后的设置不再生效。

https://blog.csdn.net/qq_24367797/article/details/118581970