about No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' error?

menyifang / DCT-Net

Official implementation of "DCT-Net: Domain-Calibrated Translation for Portrait Stylization", SIGGRAPH 2022 (TOG); Multi-style cartoonization

Apache License 2.0

734 stars 74 forks source link

Open zhanghongyong123456 opened 1 year ago

zhanghongyong123456 commented 1 year ago

Need change this code ,set
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
屏幕截图_9