Pytracking+ubuntu20.04+nvidia RTX 3060成功运行

Pytracking+ubuntu20.04+nvidia RTX 3060成功运行,第1张

参考博文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~

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/570792.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-09
下一篇 2022-04-09

发表评论

登录后才能评论

评论列表(0条)

保存