直接上结论
- 指定的GPU序号和真实的序号并不是一一对应。
- 代码中指定优先级高于shell指定
- Pycharm全局搜索代码有没有其他地方有指定GPU
CTRL+SHIFT+F (Windows) Command+Shift+F (Mac)
炼丹过程中,尝试了两种方式在代码中指定GPU,发现都没有正确指定,佛了。。全局搜索项目代码,其他没有任何地方指定了GPU。
方式1 – shell指定:
CUDA_VISIBLE_DEVICES=0 python code.py
方式2 – 代码内部指定:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"