摘要: 1. 获取GPU model.to('cuda:0') 2. 代码跑起来之后中途卡死:调用到其他显卡 则会卡死 3. 报错:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED (1)检查是否放在GUP上 (2)batch_size. num 阅读全文
posted @ 2023-03-02 11:17 helloWorldhelloWorld 阅读(316) 评论(0) 推荐(0) 编辑