镜像可以打包cuda环境吗 cuda国内镜像

转载

智能探索者 2024-03-26 05:12:20

文章标签 镜像可以打包cuda环境吗深度学习 ubuntu 服务器 CUDA 文章分类 云原生云计算

初始环境：ubuntu18.04.4 LTS，已经安装对应的显卡驱动，安装过程中不建议更换apt-get国内镜像源

一、安装CUDA10.2

1、在nvidia 官网上下载对应的CUDA 版本，我们这里选择CUDA10.2 的版本，链接https://developer.nvidia.com/cuda-toolkit-archive。如图1.1 所示选择对应的系统版本和CUDA 版本下载至本地目录中，以下载到/home/einai/Downloads 为例。
注：可先下载到本地，然后在同一局域网中通过scp –r 指令传输到服务器的指定文件夹。

镜像可以打包cuda环境吗 cuda国内镜像_服务器

图1.1 2、下载完成后，在服务器的命令窗口中依次输入cdcd Downloadsls，查看是否存在cuda_10.2.89_440.33.01_linux.run 的文件。在该文件的目录下输入指令:

sudo sh cuda_10.2.89_440.33.01_linux.run

进行安装，输入本服务器的密码后会出现如图1.2 的画面。在最后一行输入accept 按下enter键，出现如图1.3 的画面。

镜像可以打包cuda环境吗 cuda国内镜像_ubuntu_02

图1.2

镜像可以打包cuda环境吗 cuda国内镜像_ubuntu_03

图1.3

如图1.3 所示，此时已经进入的CUDA 的安装界面，其中第一个选项Diver 是显卡驱动，因为本服务器已经安装显卡驱动，所以不再勾选，如果勾选安装会出现bug，慎重！其余4项默认安装，选择好后将光标移至’Install’处，按下enter 键，此时已经开始安装，切勿在命令窗口按下其它按键，等待即可。

3、安装完毕后，输入nvcc –V，如果出现下图1.4 所示，表明此时可能没有添加CUDA 的环境变量。在命令窗口依次输入：

cd gedit ~/.bashrc，

查看文件中是否有以下语句：

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

如果没有，添加进去，添加完毕后保存如图1.5 所示。此时依次输入指令:

source ~/.bashrc
nvcc –V

如果出现图1.6 所示画面，则表明CUDA 安装成功。

镜像可以打包cuda环境吗 cuda国内镜像_镜像可以打包cuda环境吗_04

图1.4

镜像可以打包cuda环境吗 cuda国内镜像_CUDA_05

图1.5

二、安装Cudnn

1、下载CUDA10.2 对应的Cudnn 版本，链接：https://developer.nvidia.com/rdp/cudnn-download。选择Cudnn8.0.0 版本进行下载，如图2.1 所示，选择圈出的三个deb 文件，高版本Cudnn 不建议使用tar 文件安装。这里依旧下载到/home/einai/Downloads 目录下。

镜像可以打包cuda环境吗 cuda国内镜像_镜像可以打包cuda环境吗_06

图2.1

2、进入到deb 文件的目录，在此路径下依次输入以下指令，顺序不能颠倒！

sudo dpkg -i libcudnn8_8.0.0.180-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.0.180-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-doc_8.0.0.180-1+cuda10.2_amd64.deb

3、(1)上述步骤已经完成Cudnn 的安装，现在需要验证Cudnn 是否安装成功。在窗口输入以下指令