参考博文pytracking ubuntu安装
问题描述:上面链接的参考博客写得很好,但是由于我显卡是3060,只支持cuda11及以上版本,所以当我使用pytorch官网的指令安装新版的pytorch,发现在运行demo的时候会出现以下报错。
THC/THC.h: No such file or directory
其原因在于新版的torch与以及删除这个包了,官方pytracking中使用的pytorch;还停留在torch1.4.0。
按照置顶博客安装好python 3.7的虚拟环境之后,就立刻在pytorch离线下载的地方下载下面这两个包,并使用pip安装。
(虽然这里torch对应的是cuda11.0,但是懒得重新装11.0就先试一试11.3能不能运行。
)
pip install torch-1.7.0+cu110-cp37-cp37m-linux_x86_64.whl
pip install torchvision-0.8.0-cp37-cp37m-linux_x86_64.whl
怕做无用功,所以安装完成以后,首先启动python环境验证torch能不能调用cuda
conda activate pytracking
python
import torch
torch.cuda.is_available()
如果这里成功返回True,就按照置顶博客继续安装其他的地方,就可以运行了!!!
在此感谢强强这个超级无敌科研小助手MUA~
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)