【工具】——vscode python指定gpu id
点击左边的运行/ 或者用快捷键 ctrl+shift+d——添加launch.json在launch.json中添加CUDA_VISIBLE_DEVICES:1{// 使用 IntelliSense 了解相关属性。// 悬停以查看现有属性的描述。// 欲了解更多信息,请访问: https://go.microsoft.com/fwlink/?linkid=830387"version": "0.2
·
- 点击左边的运行/ 或者用快捷键 ctrl+shift+d——添加launch.json
- 在launch.json中添加CUDA_VISIBLE_DEVICES:1
{
// 使用 IntelliSense 了解相关属性。
// 悬停以查看现有属性的描述。
// 欲了解更多信息,请访问: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"name": "Python: 当前文件",
"type": "python",
"request": "launch",
"program": "${file}",
"console": "integratedTerminal",
"env":{
"CUDA_VISIBLE_DEVICES":"1"
}
}
]
}
- 结果
可以通过nvidia-smi看现存占用情况
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
已为社区贡献26条内容
所有评论(0)