使用VScode使用建立容器时默认的方式是不使用GPU的,我们需要设置一下settings.json

点击IMAGES栏的问号,选择Edit settings

选择后,找到run 和run interactive

在setting中设置,添加--gpus all

这样在对IMAGES栏的容器构建容器时,选择run interactive就可以使用gpu了
版权声明:本文为weixin_37707670原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
使用VScode使用建立容器时默认的方式是不使用GPU的,我们需要设置一下settings.json

点击IMAGES栏的问号,选择Edit settings

选择后,找到run 和run interactive

在setting中设置,添加--gpus all

这样在对IMAGES栏的容器构建容器时,选择run interactive就可以使用gpu了