首页 > Python资料 博客日记
低版本CUDA安装/多版本切换/用户级CUDA安装
2024-05-08 19:00:04Python资料围观1008次
前言:我想要安装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系统版本和架构
uname
lsb_release -a
CUDA版本和驱动版本的对照表
参照官网: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上进行切换。
因此显卡驱动总是越高越好,依赖关系为:
- 操作系统版本最优先
- 显卡驱动次之
- CUDA再次之+Cudnn
- 最后是python库,如pytorch版本
下载CUDA 10.2
运行如下wget命令在服务器安装cuda的目录下:
将下载好的文件执行以下操作:
- 进入到该目录下,使用
chmod 755 cuda_10.2.89_440.33.01_linux.run
更改文件的执行权限。 - 不是管理员用户,因此无法使用sudo安装,直接执行以下命令即可
sh cuda_10.2.89_440.33.01_linux.run
安装过程
- 输入accept接收协议
- 选择只安装CUDA Toolkit,选择
Option
回车进入
- 修改安装路径
前提:如下使用的两个目录需要先创建
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,已经安装成功
安装低版本PyTorch 1.7.1
下载网址:https://pytorch.org/get-started/previous-versions/
pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
直接运行如上命令即可安装成功
Ref
标签:
相关文章
最新发布
- 光流法结合深度学习神经网络的原理及应用(完整代码都有Python opencv)
- Python 图像处理进阶:特征提取与图像分类
- 大数据可视化分析-基于python的电影数据分析及可视化系统_9532dr50
- 【Python】入门(运算、输出、数据类型)
- 【Python】第一弹---解锁编程新世界:深入理解计算机基础与Python入门指南
- 华为OD机试E卷 --第k个排列 --24年OD统一考试(Java & JS & Python & C & C++)
- Python已安装包在import时报错未找到的解决方法
- 【Python】自动化神器PyAutoGUI —告别手动操作,一键模拟鼠标键盘,玩转微信及各种软件自动化
- Pycharm连接SQL Sever(详细教程)
- Python编程练习题及解析(49题)
点击排行
- 版本匹配指南:Numpy版本和Python版本的对应关系
- 版本匹配指南:PyTorch版本、torchvision 版本和Python版本的对应关系
- Anaconda版本和Python版本对应关系(持续更新...)
- Python 可视化 web 神器:streamlit、Gradio、dash、nicegui;低代码 Python Web 框架:PyWebIO
- 相关性分析——Pearson相关系数+热力图(附data和Python完整代码)
- Python与PyTorch的版本对应
- Windows上安装 Python 环境并配置环境变量 (超详细教程)
- Python pyinstaller打包exe最完整教程