Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count

auspicious3000 / autovc

AutoVC: Zero-Shot Voice Style Transfer with Only Autoencoder Loss

https://arxiv.org/abs/1905.05879

MIT License

983 stars 207 forks source link

Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count #42

Closed j2l closed 4 years ago

j2l commented 4 years ago

Running conversion.ipynb for the first time, I got the runtime error in the title. I fixed it using; https://github.com/computationalmedia/semstyle/issues/3 adding ,map_location='cuda:0' to torch.load Not sure why though.