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

[pouramini]

Setting

'CUDA_VISIBLE_DEVICES=0,1'

at the beginning of file can solve the problem…