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
的过程,我懂得了两个道理:
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/
转载文章受原作者版权保护。转载请注明原作者出处!