model.to(torch::kHalf); } model.to(device);//GPU版异常,可能模型并没有完全放到GPU上 2、解决方法 highlighter- dns model = torch::jit::load(ptFile, torch::kCUDA); 参考:https://github.com/pytorch/pytorch/issues/19302 1、原因 __EOF__...
PyTorch version: 1.10.1 Is debug build: False CUDA used to build PyTorch: None ROCM used to build PyTorch: N/A OS: Ubuntu 20.04.2 LTS (x86_64) GCC version: (Ubuntu 9.3.0-17ubuntu1~20.04) 9.3.0 Clang version: Could not collect CMake version: Could not collect Libc version: glibc-...