在配置远程服务器环境中
pip install git+https://github.com/facebookresearch/pytorch3d.git@stable
时报错
报错显示:
RuntimeError:
The detected CUDA version (12.2) mismatches the version that was used to compile
PyTorch (11.8). Please make sure to use the same CUDA versions.
原因:
由于服务器上的CUDA环境为12.2,anaconda环境cuda版本为11.8,两者跨越一个大版本容易不兼容,出现安装不成功的情况。
但是pytorch官网(PyTorch)目前只更新到cuda12.1。
查找经验贴后,发现可以将cuda降级,但实在是太麻烦了。后来直接将torch更新到了最新的版本,实践证明是可以用的。
先将原来的torch卸载:
pip uninstall torch
再将新版本安装:
# CUDA 12.1
pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121