Linux服务器非root用户下安装CUDA11.1和cudnn到指定目录

(适用于服务器上管理员已安装好显卡驱动或已安装的CUDA版本无法满足自己要求)

1、准备工作

首先查看自己服务器系统的信息

uname -a 
lsb_release -a

这是我的服务器的信息
在这里插入图片描述
然后根据版本去官网选择对应版本下载CUDA
Linux服务器非root用户下安装CUDA11.1和cudnn到指定目录_第1张图片
接着下载cudnn(网址),选择CUDA11.1对应的版本 (需登录nvidia账号,无法用wget,我只能采取本地下载上传到服务器)。

选择 “cuDNN Library for Linux”,下载下来是一个tgz的压缩包。

最后,提前建好cuda的自定义目录,我这里以当前用户名目录为例,安装cuda-11.1:

cd /home/zhuang/
mkdir cuda-11.1
mkdir cuda-11.1/mylib

2、安装CUDA

bash cuda_11.1.1_455.32.00_linux.run

选择accept 后,更改安装选项(‘X’ 代表选中,这里我只安装CUDA Toolkit和CUDA Samples,一般不搞CUDA编程单跑深度学习就够了):

┌──────────────────────────────────────────────────────────────────────────────┐
| CUDA Installer                                                               │
│ - [ ] Driver                                                                 │
│      [ ] 455.32.00                                                           │
│ + [X] CUDA Toolkit 11.1                                                      │
│   [X] CUDA Samples 11.1                                                      │
│   [ ] CUDA Demo Suite 11.1                                                   │
│   [ ] CUDA Documentation 11.1                                                │
│   Options                                                                    │
│   Install                                                                    │

选中Options:

┌──────────────────────────────────────────────────────────────────────────────┐
│ Options                                                                      │
│   Driver Options                                                             │
│   Toolkit Options                                                            │
│   Samples Options                                                            │
│   Library install path (Blank for system default)                            │
│   Done                                                                       │
│                                                                              │

先更改Toolkit Options (/usr这种非用户目录的都要去掉,我这里全去掉了,另外进入 Change Toolkit Install Path设置cuda安装到自己具有写入权限的路径(提前建好),我这里是“/home/zhuang/cuda-11.1/”):

┌──────────────────────────────────────────────────────────────────────────────┐
│ CUDA Toolkit                                                                 │
│   Change Toolkit Install Path                                                │
│   [ ] Create symbolic link from /usr/local/cuda                              │
│ - [ ] Create desktop menu shortcuts                                          │
│      [ ] Yes                                                                 │
│      [ ] No                                                                  │
│   [ ] Install manpage documents to /usr/share/man                            │
│   Done                                                                       │
│                                                                              │
┌──────────────────────────────────────────────────────────────────────────────┐
│ Change Toolkit Install Path                                                  │
│ /home/zhuang/cuda-11.1/                                                    │
│                                                                              │

Samples Options安装类似,目录为/home/zhuang/cuda-samples

做完Done,回到Options菜单, 更改Library install path (不改不行,它会偷偷写入/var/lib)

┌──────────────────────────────────────────────────────────────────────────────┐
│ Library install path (Blank for system default)                              │
│ /home/zhuang/cuda-11.1/mylib/

上述两个自定义的目录最好提前自己手动建好。安装成功后,会出个summary。

更改环境变量:

vim ~/.bashrc

在尾部添加(将cuda的各种库目录添加到环境变量):

# CUDA
export PATH="/home/zhuang/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/home/zhuang/cuda-11.1/lib64:/home/zhuang/cuda-11.1/mylib/lib64:$LD_LIBRARY_PATH"

刷新环境变量:

source ~/.bashrc

测试CUDA:
Linux服务器非root用户下安装CUDA11.1和cudnn到指定目录_第2张图片

3、安装cudnn

具体目录需要根据情况修改,我这里CUDA安装到了 ~/cuda-11.1 内

tar -zxvf cudnn-11.1-linux-x64-v8.0.4.30.tgz cd cuda
cd cuda # 此处进入cudnn解压的目录

cp ./include/cudnn.h ~/cuda-11.1/include
cp ./lib64/libcudnn* ~/cuda-11.1/lib64

chmod a+r ~/cuda-11.1/include/cudnn.h ~/cuda-11.1/lib64/libcudnn*

到此cuda的安装应该算是完成了.

你可能感兴趣的:(常用工具,服务器,linux,运维)