淘先锋技术网

首页 1 2 3 4 5 6 7

1、提前准备两个包,放在home目录下:

cuda_10.0.130_410.48_linux.run

cudnn-10.0-linux-x64-v7.6.4.38.tgz

 2、先卸载系统中旧版本cuda9.0,命令如下:

sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl

 3、卸载之后,会发现 /usr/local/cuda-9.0目录下仍然有文件存在,将cuda9.0版本文件全部删除干净,命令如下:

sudo rm -rf /usr/local/cuda-9.0

 4、安装cuda10.0,命令如下:

sudo sh cuda_10.0.130_410.48_linux.run

一直按住回车键,直到服务条款为100%时,执行分别如下命令:

accept

no(注:不安装驱动),其他都yes

5、 安装完成后需要设置环境变量:

5.1、打开.bashrc文件,命令为:sudo gedit ~/.bashrc

5.2、在打开的文件末尾,添加以下三行:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0

export PATH=$PATH:/usr/local/cuda-10.0/bin
5.3、保存文件后输入命令更新文件:source ~/.bashrc

5.4、测试cuda10.0是否安装成功,重新打开终端输入:nvcc -V 

可以查看cuda版本型号信息,如果查询失败,重启电脑再试一下。

6、安装cudnn, 解压cudnn-10.0-linux-x64-v7.6.4.38.tgz,直接打开终端执行如下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

检查cudnn是否安装成功, 命令如下:

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

如果出现如下字样,说明安装成功

#define CUDNN_MAJOR 7

#define CUDNN_MINOR 6

#define CUDNN_PATCHLEVEL 4

--

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

注意事项!!!!

1、关于cuda卸载和升级都是同样的操作 ,主要分为两步:一个是卸载旧版本,另一个是安装新版本,关键在于不同旧版本的安装位置有所不同,需要我们去寻找一下文件的具体位置,不要盲目复制粘贴代码。

2、可以对照其他博客,具体问题具体分析。