What your version of pyTorch?

clcarwin / sphereface_pytorch

A PyTorch Implementation of SphereFace.

MIT License

714 stars 172 forks source link

What your version of pyTorch? #40

Open 312shan opened 6 years ago

312shan commented 6 years ago

Hello , What your version of pyTorch? I run train model in cuda==0.4.1 and get RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/THCGeneral.cpp:663

chaochao12345lala commented 4 years ago

python2.7 pytorch0.4.1-cuda8.0