使用ubuntu cuda11.2安装pytorch
在使用镜像新建了一个cuda11.2-python3.9
容器配置环境的过程中需要安装PyTorch。一开始我直接使用 pip install torch
来进行安装,但是运行程序时出现报错:
RuntimeError: The NVIDIA driver on your system is too old (found version 11020). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.
这个错误表明目前系统上安装的 NVIDIA 驱动程序版本太旧,不满足 PyTorch 对 cuda驱动程序的要求。根据提示我们可以选择更新NVIDIA 驱动,或者安装适当版本的PyTorch。这里我选择了更新PyTorch。
首先通过 nvcc -V
指令查看cuda版本,得知为版本11.2
在官方安装pytorch2.0的文档中,最低cuda版本为11.8.因此我们需要安装一个旧版pytorch
在官方的旧版pytorch与cuda版本对应中可以找到支持版本11.2以下的最高版本是v1.12.1,通过安装指令:
# CUDA 10.2
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
本文来自博客园,作者:落魄统计佬,转载请注明原文链接:https://www.cnblogs.com/tungsten106/p/install_pytorch_cuda112_ubuntu.html
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· spring官宣接入deepseek,真的太香了~