Pytorch 版本问题 THC/THC.h: No such file or directory
该问题发生于安装
c
语言扩展时。
这个问题我经常遇见,也是因为我之前不关心pytorch
版本造成的坏习惯。
🙋 张同学 zhangruiyuan@zju.edu.cn 有问题请联系我呀~
一、我是怎么尝试解决这个问题的
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
的过程,我懂得了两个道理: