文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

低版本CUDA安装/多版本切换/用户级CUDA安装

2023-09-18 09:56

关注

前言:我想要安装Torch V1.7.1,根据版本信息只有cuda10.2和11.0。但是,本地安装的CUDA版本为11.6和11.2,都不能满足需求。因此需要降低CUDA版本为10.2

CUDA版本和显卡驱动

nvidia-smi

目前CUDA版本为11.6,如下图所示,显卡驱动为510.47.03(根据下面的对照表,降级成CUDA10.2没问题)
在这里插入图片描述

ubuntu系统版本和架构

unamelsb_release -a

请添加图片描述

参照官网:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

在这里插入图片描述
如上图所示,像CUDA10.2版本,需要440.33以上的显卡驱动。例如,如果你通过nvidia-smi命令获得的显卡驱动号大于440.33,你可以将CUDA 11降到10.2是完全没有问题的。而如果你的显卡驱动号是396.26,那么你只能在CUDA7/8/9上进行切换。

因此显卡驱动总是越高越好,依赖关系为:

下载网址:https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

运行如下wget命令在服务器安装cuda的目录下:
在这里插入图片描述
将下载好的文件执行以下操作:

安装过程

  1. 输入accept接收协议
    请添加图片描述
  2. 选择只安装CUDA Toolkit,选择Option回车进入
    请添加图片描述
  3. 修改安装路径

前提:如下使用的两个目录需要先创建
1) /home/[用户名]/cuda-10.2/
2)/home/[用户名]/cuda-10.2/mylib/

先选择Toolkit Options回车进入
在这里插入图片描述

选择Change Toolkit Install Path回车,输入自定义的CUDA安装目录:

/home/[用户名]/cuda-10.2/

提前手动创建

请添加图片描述

请添加图片描述更改Library install path
请添加图片描述

/home/[用户名]/cuda-10.2/mylib

提前手动创建

安装成功

请添加图片描述
此时nvcc还是显示原来的CUDA版本,还需要修改本地文件,如下:

修改环境配置文件

修改命令相关参数,执行如下命令:

vim ~/.bashrc

在最下方添加刚刚安装cuda的路径:

export PATH="/home/[用户名]/cuda-10.2/bin:$PATH"export LD_LIBRARY_PATH="/home/[用户名]/cuda-10.2/lib64:/home/[用户名]/cuda-10.2/mylib/lib64:$LD_LIBRARY_PATH"

之前的CUDA路径可以注释掉,方便之后使用
请添加图片描述

保存之后,执行以下命令,使配置生效:

source~/.bashrc

查看效果:nvcc -V
请添加图片描述
显示CUDA 10.2,已经安装成功

下载网址:https://pytorch.org/get-started/previous-versions/
在这里插入图片描述

pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2

直接运行如上命令即可安装成功

  1. 用户级别CUDA安装教程:https://blog.csdn.net/qq_43800119/article/details/130149338?spm=1001.2014.3001.5501

来源地址:https://blog.csdn.net/qq_43800119/article/details/130566105

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯