在容器里，模型运行到model.cuda()这一步很久不动，直到被kill。
解决方案：
查看容器里cuda为10.2版本，但是torch安装的是0.4.0版本，所以我重新安装为匹配的版本：
pip3 install torch==1.6.0 torchvision==0.7.0
这样就解决了。
容器里 pytorch model.cuda()卡死
最新推荐文章于 2025-06-22 17:06:13 发布
本文介绍了解决模型在容器环境中因CUDA版本与PyTorch版本不匹配导致的问题。通过调整PyTorch及其相关库的版本以适配现有的CUDA版本10.2，成功解决了模型运行停滞的问题。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
325
到【灌水乐园】发言
