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后成功运行:
文章出处登录后可见!
已经登录?立即刷新