search

chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
MIT License
11.99k stars 3.73k forks source link

第九章使用GPU时出现runtime error #198

Closed zhengyyi closed 4 years ago

zhengyyi commented 4 years ago

RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/THC/THCBlas.cu:258 请问应该如何解决？

zhengyyi commented 4 years ago

t.cuda.current_device() t.cuda._initialized = True