使用ubuntu cuda11.2安装pytorch
  bVtmceYJfDgV 2023年12月15日 33 0

在使用镜像新建了一个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
【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年12月15日 0

暂无评论

推荐阅读
EM
  YS70eVTgu2rK   6天前   16   0   0 机器学习
HMM
  YS70eVTgu2rK   6天前   16   0   0 机器学习
bVtmceYJfDgV