1 下载CUDA Toolkit

  1. 进入链接https://developer.nvidia.com/cuda-downloads?
  2. 选择对应的环境在这里插入图片描述
  3. 执行网站自动生成的第一条命令行去下载Toolkit,请忽略第二条命令
    在这里插入图片描述

2 安装CUDA

  1. 运行你所下载的安装包,形如: sh cuda_11.1.0_455.23.05_linux.run
  2. 首先要明白,Toolkit里面不仅有CUDA,还有个驱动包,所以在安装的时候注意,不要安装驱动(当然,没有root权限也是装不上去的)。
  3. 仔细阅读安装提示(都很短的)。一般就是:1条款;2是否安装驱动;3是否安装CUDA;4指定安装路径

2 配置用户的环境变量

  1. 修改个人用户的环境变量
vim ~/.bashrc
  1. 末尾添加环境变量(下面的路径是我的路径,请修改成你的)
export PATH=$HOME/cuda/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda/lib64/
  1. 运行source ~/.bashrc使环境变量生效

3 检查是否生效

  1. 运行nvidia-smi 查看显卡驱动运行状态
  2. 运行nvcc -V 查看cuda-toolkit安装是否成功

4 安装CuDNN

  1. 上https://developer.nvidia.com/cudnn下载CuDNN库
  2. 解压CuDNN压缩包
  3. 根据安装官网的教程,把下载下来的CuDNN中的头文件和库文件复制,剪贴,或移动到教程指定的路径下(当然是放到你本地的CUDA里面)。
  4. 官网教程也会要求你使用chmod设定文件权限,造做就行
Logo

更多推荐