一般我们指定GPU训练时,用到的是:
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"
今天遇到一个有趣的问题,在我利用以上语句进行指定GPU的时候,上面语句失效,及无论我如何进行指定,都无法实现仅利用‘1,2’块GPU操作,总是利用到原来的0块GPU。
**解决方法:于是我将os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"代码放到了import os后面解决了!!!!!
即:如下
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"
语句不在失效~!!!!!!不知道什么原因。

本文描述了一种在深度学习中常见的问题,即使用os.environ[CUDA_VISIBLE_DEVICES]指定GPU时出现的失效情况。通过调整代码顺序,即将该语句放置在import os之后,成功解决了GPU指定问题。
2444

被折叠的 条评论
为什么被折叠?



