跨境派

跨境派

跨境派,专注跨境行业新闻资讯、跨境电商知识分享!

当前位置:首页 > 工具系统 > ERP系统 > 【Ubuntu 20.04LTS系统】安装CUDA11.8、cuDNN,可进行CUDA版本切换

【Ubuntu 20.04LTS系统】安装CUDA11.8、cuDNN,可进行CUDA版本切换

时间:2024-04-09 10:25:41 来源:网络cs 作者:康由 栏目:ERP系统 阅读:

标签: 版本  系统  安装 

Ubuntu 20.04LTS系统安装CUDA11.8、cuDNN,可进行CUDA版本切换

1. 更改为清华源并更新软件列表和依赖项2. 安装CUDA3. 安装cuDNN4. CUDA版本切换

1. 更改为清华源并更新软件列表和依赖项

https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/

# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal main restricted universe multiverse# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal main restricted universe multiversedeb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-updates main restricted universe multiverse# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-updates main restricted universe multiversedeb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-backports main restricted universe multiverse# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-backports main restricted universe multiverse# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-security main restricted universe multiverse# # deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-security main restricted universe multiversedeb http://security.ubuntu.com/ubuntu/ focal-security main restricted universe multiverse# deb-src http://security.ubuntu.com/ubuntu/ focal-security main restricted universe multiverse

更新软件列表和必要的依赖项

sudo apt-get update# 更新软件列表sudo apt-get install g++# 下载g++编译器sudo apt-get install gcc# 下载gcc编译器sudo apt-get install make# 下载GNU Make编译器sudo apt-get install initramfs-tools   # 下载安装initramfs-tools

2. 安装CUDA

步骤一: 下载CUDA安装包

进行CUDA和cuDNN的选择,也可以直接根据官方推荐进行下载安装。从Nvidia官网下载CUDA https://developer.nvidia.com/cuda-downloads
下方链接,选择更多版本 https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述接下来根据提示进行下载、安装
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.runsudo sh cuda_11.8.0_520.61.05_linux.run

步骤二: 安装CUDA安装包过程中的选择

Do you accept the above EULA? (accept / decline / quit):
是否接受最终用户许可协议,输入accept
在这里插入图片描述

回车键进行勾选,X就是选中,没有X就是没有选中,把安装驱动进行取消。之后向下键,回车确认
在这里插入图片描述

最后点击 install

出现了警告信息,对安装无影响在这里插入图片描述

截图日志:

wit622@622:~/cuda$ sudo sh cuda_11.8.0_520.61.05_linux.run[sudo] wit622 的密码: ============ Summary ============Driver:   Not SelectedToolkit:  Installed in /usr/local/cuda-11.8/Please make sure that -   PATH includes /usr/local/cuda-11.8/bin -   LD_LIBRARY_PATH includes /usr/local/cuda-11.8/lib64, or, add /usr/local/cuda-11.8/lib64 to /etc/ld.so.conf and run ldconfig as rootTo uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-11.8/bin***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 520.00 is required for CUDA 11.8 functionality to work.To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:    sudo <CudaInstaller>.run --silent --driverLogfile is /var/log/cuda-installer.log

也可以进入日志进行查看(位置如下):

 /var/log/cuda-installer.log

步骤三: 配置CUDA环境

sudo  vim ~/.bashrc
在bashrc文件最下方,添加下入代码(ps:这边需要注意cuda的版本,版本不同,路径的命名需修改)
export PATH=$PATH:/usr/local/cuda-11.8/binexport LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
更新环境
source ~/.bashrc
测试CUDA是否安装成功
nvcc -V

输出下述结果,表示安装成功:

wit622@622:~/cuda$ nvcc -Vnvcc: NVIDIA (R) Cuda compiler driverCopyright (c) 2005-2022 NVIDIA CorporationBuilt on Wed_Sep_21_10:33:58_PDT_2022Cuda compilation tools, release 11.8, V11.8.89Build cuda_11.8.r11.8/compiler.31833905_0

3. 安装cuDNN

步骤一:下载cuDNN包

https://developer.nvidia.com/rdp/cudnn-archive

例如这边选择了Download cuDNN v8.8.1 (March 8th, 2023), for CUDA 11.x点击Local Installer for Linux x86_64 (Tar)即可下载压缩包
在这里插入图片描述将压缩包,放入自定义路径后,输入命令进行解压
tar -xvf cudnn-linux-x86_64-8.8.1.3_cuda11-archive.tar.xz 
解压后,输入命令,讲cuDNN对应文件拷贝至CUDA指定路径。
wit622@622:~/下载$ cd cudnn-linux-x86_64-8.8.1.3_cuda11-archive/wit622@622:~/下载/cudnn-linux-x86_64-8.8.1.3_cuda11-archive$ lsinclude  lib  LICENSE
sudo cp include/cudnn*.h /usr/local/cuda-11.8/includesudo cp lib/libcudnn* /usr/local/cuda-11.8/lib64sudo chmod a+r /usr/local/cuda-11.8/include/cudnn*.h /usr/local/cuda-11.8/lib64/libcudnn*

4. CUDA版本切换

因为后续部分项目,用的库对应的CUDA不同,无需进行修改,修改环境CUDA路径即可。例如需要11.1的CUDA,可通过修改bashrc进行修改

sudo vim ~/.bashrc

将原先的cuda-11.8注释掉,添加cuda-11.1新的环境设置,即可

# cuda-11.8# export PATH=$PATH:/usr/local/cuda-11.8/bin# export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64# cuda-11.1export PATH=$PATH:/usr/local/cuda-11.1/binexport LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64

本文链接:https://www.kjpai.cn/news/2024-04-09/155867.html,文章来源:网络cs,作者:康由,版权归作者所有,如需转载请注明来源和作者,否则将追究法律责任!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。

文章评论