[Debug] Pytorch 版本问题 THC/THC.h: No such file or directory

Pytorch 版本问题 THC/THC.h: No such file or directory

该问题发生于安装c语言扩展时。
这个问题我经常遇见，也是因为我之前不关心pytorch版本造成的坏习惯。
张同学 [email protected] 有问题请联系我呀~

一、我是怎么尝试解决这个问题的

1.1 当我在默认安装的pytorch中执行时，会遇到上述问题。

# 在我写这个文章的时候，pytorch为1.11.0
conda install pytorch torchvision torchaudio cpuonly -c pytorch

我没有找到我的日志图了，emmm，反正确实是输出提示THC/THC.h: No such file or directory。

1.2 然后我尝试了一下下面这些版本，但是会遇到另外两个问题。

i) 问题一

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

ii) 问题二

我尝试了pytorch为1.9.1和1.8.1的情况，遇到了下面的这个问题

conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c pytorch -c conda-forge
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge

1.3 接着我去检查了一下torch的版本，发现目前下载的是cpu的版本？真实搞不懂为什么，所以我打算重新构建环境。

重新构建环境，并重新安装1.10.0版本

1.4 最终我的问题得以解决！！！ 开心

二、结论

经过这次Debug的过程，我懂得了两个道理：

1. pytorch版本是个不容忽视的问题，很多bug就是因为这个产生的，要注意观察版本号和是否是gpu版本。

2. 不要相信在conda环境中安装另一个版本的pytorch，很可能出现问题，建议还是删除现有的环境再重新安装pytorch。