ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

Linux 安装多版本CUDA

2022-04-21 17:32:37  阅读:386  来源: 互联网

标签:libcudnn CUDA sudo usr 版本 Linux local cuda


安装CUDA

同样的在运行不同代码的时候,需要的环境不同,所以需要不同的GPU环境,Cuda的需要自然也就发生改变 特别是关于tensorflow的运行中。

首先我们需要在nvidia的官网上找到我们所需要的cuda版本,接下来可以根据各自的实际要求来选择

当前环境配置:Ubuntu20.04 gcc-5, gcc-8, gcc-9, cuda 11.3

需求安装cuda 9

屏幕截图 2022-04-20 103810

然后运行安装这个runfile sudo sh xxx.run

注意不要安装英伟达驱动 因为之前已经有一个cuda环境 安装过了

image-20220420121744526

然后修改cuda的环境变量 vim ~/.bashrc (如果环境变量使用的是cuda软连接 就可以不用修改环境变量了)

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

source ~/.bashrc保存环境变量,使配置生效

进入cd /usr/local

image-20220420121929018

发现cuda其实一个软连接,所以以后我们需要切换cuda版本的时候,只需要修改这个软连接的指向就可以了

sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.5 /usr/local/cuda

可以利用nvcc -V来查看当前CUDA的版本

image-20220421104236667

安装Cudnn

当然有的时候在使用CUDA的时候,还需要使用Cudnn,所以我们需要在对应的CUDA环境内安装我们需要的Cudnn

Cudnn的下载地址,下载的时候需要先登陆一下

然后根据自己环境和CUDA来选择下载的版本 我这里选择是下方的版本

屏幕截图 2022-04-21 102906

対以下操作进行修改:

tar -xzvf cudnn-9.0-linux-x64-v7.6.5.32.tgz
sudo cp cuda-9.0/include/cudnn.h  /usr/local/cuda-9.0/include
sudo cp cuda-9.0/lib64/lib*  /usr/local/cuda-9.0/lib64


sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.6.5 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h   /usr/local/cuda-9.0/lib64/libcudnn*

标签:libcudnn,CUDA,sudo,usr,版本,Linux,local,cuda
来源: https://www.cnblogs.com/young978/p/16169803.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有