ImportError:/../libc10 cuda.so: undefined symbol: cudaMempoolsetAttribute, version libcudart.so.11.0
“ImportError: /root/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/lib/libc10 cuda.so: undefined symbol: cudaMempoolsetAttribute, version libcudart.so.11.0”
如何解决“ImportError: /root/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/lib/libc10 cuda.so: undefined symbol: cudaMempoolsetAttribute, version libcudart.so.11.0”问题

我在Jupyter Notebook中运行python代码时,需要用到torch。但是出现了这种问题,“ImportError: /root/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/lib/libc10 cuda.so: undefined symbol: cudaMempoolsetAttribute, version libcudart.so.11.0”,在一番查证发现,是因为我的环境与代码需要的环境【版本】不兼容。
根据网上的一些资料,我查到我的CUDA Version是12.0
命令:nvidia-smi
命令:pip list
命令:python --version
但是报错显示,没有找到11.0版本的文件。所以第一想法是,降级。把我的版本降到11.0.但是并没有效果。无论怎样,始终是CUDA Version:12.0
于是,又仔细看了我需要运行代码的环境,要求的环境是
-python3.8.13,
-torch== 1.12.1+cu116
- torchaudio== 0.12.1+cu116
- torchvision== 0.13.1+cu116
于是直接下载安装了对应的版本:
【网址:https://pytorch.org/get-started/previous-versions/】
命令:pip install torch== 1.12.1+cu116 torchvision== 8.13.1+cu116 torchaudi0==8.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
下载完成后重新查询:pip list
再重新运行代码:可正常运行,无报错!
总结:如果版本不正确,我们需要知道我们将运行的代码所需要的环境是怎样的,然后去官网下载对应的版本。
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐

所有评论(0)