DCT-Net
DCT-Net copied to clipboard
about No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' error?
Need change this code ,set
os.environ['CUDA_VISIBLE_DEVICES'] = '0'