本 bug 发生于 2024年10月,如果时间过了太久解决办法不一定能成功
出现 bug 的过程如下:
背景:在 anaconda prompt 输入 nvidia-smi 获得当前显卡支持的最高 cuda 版本为12.2。因为某些库的需要,安装的是 11.8 的 cuda。在 cmd 输入 nvcc -V 能看到 cuda==11.8
操作:anaconda prompt 输入 activate xxx 创建环境。然后在 pytorch.org 找到 cuda11.8 的 pytorch 安装指令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
然后弹出一系列即将安装的库,其中包括 numpy==2.0.x
结果:运行代码出现报错 ImportError: numpy.core.multiarray failed to import
解决办法:卸载numpy,重新安装numpy为指定版本:pip install numpy==1.26.4
结论:pytorch.org 给的安装指令自动安装了不合适的版本