identifier “THCudaCheck“ is undefined 的解决方法

THCudaCheck 在 pytorch 1.11.0 版本被移除了,可以看文档 https://www.exxactcorp.com/blog/Deep-Learning/pytorch-1-11-0-now-available

解决方法是将 THCudaCheck 替换成 C10_CUDA_CHECK

你可能感兴趣的:(Machine,Learning,Python,Pytorch,深度学习,pytorch,人工智能)