解决安装库时报错:RuntimeError: CUDA版本不匹配 (11.7) 与使用的版本不符…

 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后成功运行:

 

 

物联沃分享整理
物联沃-IOTWORD物联网 » 解决安装库时报错:RuntimeError: CUDA版本不匹配 (11.7) 与使用的版本不符…

发表评论