colab中遇到“THC/THC.h: No such file or directory”的解决办法

若在运行colab是遇到如下问题
fatal error: THC/THC.h: No such file or directory

问题原因:pytorch在高版本(1.11以后)将THC/THC.h文件删除了。
解决办法:降低pytorch版本,如降为1.5版本。
执行代码:!pip install torch==1.5.0 torchvision

你可能感兴趣的:(colab中遇到“THC/THC.h: No such file or directory”的解决办法)