pid = os.getbid() #获取当前进程id
os.environ['CUDA_DEVICE_ORDER'] = "PCI_BUS_ID" 让显卡编号固定住,看到的就是真实编号
os.environ["CUDA_VISIBLE_DEVICE"] = "0,1,2" 设置cuda可用的显卡
- 显示当前工作路径
os.getcwd()
- 目录中所有文件名
fileName = os.listdir(dataPath)
python中os操作
最新推荐文章于 2023-03-19 15:38:51 发布