windows / linux 安装 cuda/cudnn

cuda/cudnn安装

• ubuntu18.04安装cuda/cudnn
• • 安装cuda
  • 安装cudnn

ubuntu18.04安装cuda/cudnn

安装cuda

• 进入官方下载页面选择需要的cuda版本，cuda下载，注意可能需要登录英伟达账号。
• 选择好需要的cuda信息后，会生成安装命令：例如我需要的版本是cuda11.3.1，在命令行终端运行下面的第一个命令即可下载，下载完成后运行第二个命令进行安装。
  
  

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
sudo sh cuda_11.3.1_465.19.01_linux.run

• 安装流程及注意事项：

1. 安装前注意禁用 nouveau，验证方法：lsmod | grep nouveau 无输出说明已禁用。禁用方法如下图：
   

   

2. 执行安装脚本进行安装：sudo sh cuda_11.3.1_465.19.01_linux.run

3. 先accept，安装项里不要选择安装驱动，其他都安装。

4. 设置环境变量：在~/.bashrc文件添加如下路径，修改后注意输入：sudo source ~/.bashrc生效

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

• 检测是否安装成功：如果Result = PASS则安装成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery 

安装cudnn

• 进入官方下载地址，cudnn下载，根据cuda版本选择合适的cudnn版本进行下载，例如我选择for CUDA 11.x，cuDNN lib… for linux(x86_64)
  
  
• 下载后解压缩：tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
• 然后进行拷贝操作：主要将解压后对应的文件拷贝到之前安装的cuda里面

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

• 测试安装是否成功：查看cudnn版本 ，如果出现版本信息表示成功。

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

• 至此cuda和cudnn安装完毕！