本人配置环境:linux服务器,ubantu18,显卡驱动11.0,安装CUDA11.0和cuDNN8.2.1。

一、安装CUDA11.0

1、先多找几篇博客,了解大概的流程,避免踩坑。

2、官网下载,复制官网命令到服务器下载。https://developer.nvidia.com/cuda-toolkit-archive

 3、输入命令,再按i键,进入vim的编辑模式。

sudo vim ~/.bashrc

4、在末尾添加如下两行,注意里面的路径要换成自己的版本,再按esc键退出vim编辑模式,并输入  :wq  保存并退出

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}

 export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64\
 ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc

5、以下命令激活配置,并使用nvcc -V查看配置效果,如图表示成功。

source ~/.bashrc

 二、安装cuDNN8.2.1

1、官网下载和CUDA对应的版本 https://developer.nvidia.com/rdp/cudnn-archive

2、解压文件

tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz

3、执行如下四条命令,注意换成自己的版本。

sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64sudo chmod a+r /usr/local/cuda-11.0/include/cudnn*.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

4、使用如下命令查看cuDNN的版本号

cat /usr/local/cuda-11.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

至此安装完成。

作者:|Akmpfen|,原文链接: https://www.cnblogs.com/Cloud-king/p/17393890.html

文章推荐

Java中synchronized的优化

rust 中 str 与 String; &str &String

Linux中重定向应注意的事情

【性能】JDK和Jmeter的安装与配置

排序算法之详解冒泡排序

MySQL事务

golang 必会之 pprof 监控系列(5) —— cpu 占用率 统计原理

ChatGPT 与 Midjourney 强强联手,让先秦阿房宫重现辉煌!

sftp基本的操作

Spring Boot中如何优雅地实现异步调用?

深度解析单例模式

一次生产环境的docker MySQL故障