wls

The error is not resolved, when I create a new environment, the error is still the same .

yes. when I use 'nvcc -V' it show that my cuda version is 10.1

> I think a change to eval.py will do. Instead of loading the model from the checkpoints folder, initialize it yourself and then just add a line of training code...