赞
踩
当执行 python setup.py develop ,有如下报错:
fatal error: THC/THC.h: No such file or directory
1.11版本后,Pytorch中的THC/THC 命名空间已失效,apex 也已将其删除,但是其中的函数已移至 ATen 命名空间。
THCudaCheck函数->AT_CUDA_CHECK
注释头文件
// #include <THC/THC.h>
// #include <THC/THCDeviceUtils.cuh>
重新编译即可
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。