THCudaCheck
在 pytorch 1.11.0 版本被移除了,可以看文档 https://www.exxactcorp.com/blog/Deep-Learning/pytorch-1-11-0-now-available
解决方法是将 THCudaCheck
替换成 C10_CUDA_CHECK
THCudaCheck
在 pytorch 1.11.0 版本被移除了,可以看文档 https://www.exxactcorp.com/blog/Deep-Learning/pytorch-1-11-0-now-available
解决方法是将 THCudaCheck
替换成 C10_CUDA_CHECK