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

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

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

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

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

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

[Debug] Pytorch 版本问题 THC/THC.h: No such file or directory
重新构建环境,并重新安装 1.10.0版本
[Debug] Pytorch 版本问题 THC/THC.h: No such file or directory

; 1.4 最终我的问题得以解决!!! 开心😄

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

二、结论

经过这次 Debug的过程,我懂得了两个道理:

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

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

Original: https://blog.csdn.net/qq_36160277/article/details/124066418
Author: 文三路张同学
Title: [Debug] Pytorch 版本问题 THC/THC.h: No such file or directory

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/706891/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球