前言
我的理解安装CUDA指的是CUDA tookit,并不是单指nvidia Driver,它包括了多个组件:
- Compiler:NVCC
- Tools:分析器profiler、调试器debuggers等
- Libraries:科学库和实用程序库
- CUDA Samples:CUDA和library API的代码示例
- CUDA Driver:驱动
安装CUDA
-
输入命令nvidia-smi,查看Driver Version
Version:550.90.07 CUDA Version:12.4(CUDA toolkit不能超过这个数) -
下载CUDA:官网链接:link
-
然后选择runfile的方式安装,记住一定要取NVIDIA Driver安装,因为假设是已经有了
-
配置环境变量
vim ~/.bashrc
# 添加以下内容:
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
# 最后:
source ~/.bashrc
# 验证安装是否成功
nvcc -V
安装cuDNN
- 参考上篇文章
http://t.csdnimg.cn/UfCt3