站点图标 AI技术聚合

安装库时报错:RuntimeError: The detected CUDA version (11.7) mismatches the version that was used to…

 Ubuntu 22.04系统在安装pytorch3d库时出现了报错信息:RuntimeError: The detected CUDA version (11.7) mismatches the version that was used to compile PyTorch (10.2). Please make sure to use the same CUDA versions.

在终端查看显卡信息(指令nvidia-smi):查看CUDA版本,此处是11.7

 根据报错信息,我现在的虚拟环境下的CUDA版本(11.7)和当前的pytorch版本不匹配。

在终端查看当前虚拟环境下的pytorch版本(指令conda list),发现pytorch版本为1.11.0,和github仓库发布者要求的版本一致。

但是注意:pytorch官网安装pytorch时候的指令细节:

 刚开始我直接用第一条指令安装pytorch1.11.0,结果报错,后面注意到虽然这几条指令都是安装pytorch==1.11.0,但是最后的CUDA工具即cudatoolkit 版本有区别,从我的报错信息来看应该安装# CUDA 11.3下面的这个版本,cudatoolkit才是11这个和本地CUDA匹配的版本(.后面的细分版本无影响)

用这条指令安装pytorch后成功运行:

 

 

文章出处登录后可见!

已经登录?立即刷新
退出移动版