• 点击左边的运行/ 或者用快捷键 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看现存占用情况
    在这里插入图片描述
Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐