1 下载CUDA Toolkit
- 进入链接https://developer.nvidia.com/cuda-downloads?
- 选择对应的环境
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/e523dd310b06afa0b143acc1b0e638d1.png#pic_center)
- 执行网站自动生成的第一条命令行去下载Toolkit,请忽略第二条命令
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/6a6618638ec9f8ea2ab3356d7a35d658.png#pic_center)
2 安装CUDA
- 运行你所下载的安装包,形如:
sh cuda_11.1.0_455.23.05_linux.run
- 首先要明白,Toolkit里面不仅有CUDA,还有个驱动包,所以在安装的时候注意,不要安装驱动(当然,没有root权限也是装不上去的)。
- 仔细阅读安装提示(都很短的)。一般就是:1条款;2是否安装驱动;3是否安装CUDA;4指定安装路径
2 配置用户的环境变量
- 修改个人用户的环境变量
vim ~/.bashrc
- 末尾添加环境变量(下面的路径是我的路径,请修改成你的)
export PATH=$HOME/cuda/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda/lib64/
- 运行
source ~/.bashrc
使环境变量生效
3 检查是否生效
- 运行
nvidia-smi
查看显卡驱动运行状态 - 运行
nvcc -V
查看cuda-toolkit安装是否成功
4 安装CuDNN
- 上https://developer.nvidia.com/cudnn下载CuDNN库
- 解压CuDNN压缩包
- 根据安装官网的教程,把下载下来的CuDNN中的头文件和库文件复制,剪贴,或移动到教程指定的路径下(当然是放到你本地的CUDA里面)。
- 官网教程也会要求你使用
chmod
设定文件权限,造做就行