search

innnky / so-vits-svc

基于vits与softvc的歌声音色转换模型
GNU Affero General Public License v3.0
3.61k stars 6 forks source link

双卡机器如何指定单卡工作 #57

Open djkcyl opened 1 year ago

djkcyl commented 1 year ago

目前我是双卡 A100 80G 0卡在跑diffsinger，但我指定了1卡也会coom

image image

djkcyl commented 1 year ago

症状像是sovits吧两张显卡都吃了

TomoGaSukunai commented 1 year ago

路过围观一下壕

wing-ap commented 1 year ago

試試

import os os.environ['CUDA_VISIBLE_DEVICES'] = '1'

IMBAepsilon commented 1 year ago

或许可以直接在train里main直接指定 n_gpus=1 ?