ran out of input

[ariameetgit]

Hello,

When I am trying to run the code for reference, it showed up that:
]$ python3 synthesize.py --text "YOUR_DESIRED_TEXT" --restore_step 900000 --mode single -p config/LJSpeech/preprocess.yaml -m config/LJSpeech/model.yaml -t config/LJSpeech/train.yaml

conda/envs/FastSpeech2/lib/python3.6/site-packages/torch/cuda/init.py:52: UserWarning: CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at /opt/conda/conda-bld/pytorch_1603729128610/work/c10/cuda/CUDAFunctions.cpp:100.)
return torch._C._cuda_getDeviceCount() > 0
Traceback (most recent call last):

model = get_model(args, configs, device, train=False)
File "FastSpeech2/utils/model.py", line 20, in get_model
ckpt = torch.load(ckpt_path,map_location=torch.device('cpu'))
File "conda/envs/FastSpeech2/lib/python3.6/site-packages/torch/serialization.py", line 595, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "conda/envs/FastSpeech2/lib/python3.6/site-packages/torch/serialization.py", line 764, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
EOFError: Ran out of input

I have tried "conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 -c pytorch" to make sure the CUDA Toolkit conda packages have been downloaded. However, it still didn't work.

Is there anyone can help me solve this problem? THANKS in advance!