cuda cudnn Linux安装
时间: 2025-01-31 22:54:52 浏览: 63
### 如何在Linux上安装CUDA和cuDNN
#### 下载CUDA和cuDNN
为了开始安装过程,需先访问NVIDIA开发者网站来获取所需的软件包。对于CUDA工具包,可以选择runfile格式进行下载[^3]。
```bash
wget https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux.run
```
而对于cuDNN库,则应通过特定链接下载对应的版本压缩包[^2]:
```bash
wget https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.3.0.75_cuda12-archive.tar.xz
```
#### 安装CUDA
下载完成后,可以通过运行脚本来完成CUDA的安装操作:
```bash
sudo sh cuda_11.6.1_510.47.03_linux.run
```
这一步骤会引导用户按照提示逐步完成整个安装流程。
#### 配置环境变量
为了让系统能够识别新安装的CUDA路径,建议编辑`~/.bashrc`文件并将下面的内容追加进去以更新PATH和LD_LIBRARY_PATH环境变量设置:
```bash
export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
之后记得重新加载该配置文件使更改生效:
```bash
source ~/.bashrc
```
#### 安装cuDNN
接下来处理cuDNN部分,首先需要解压之前下载得到的tarball档案[^4]:
```bash
tar -zxvf cudnn-10.1-linux-x64-v7.5.1.10.tgz
```
随后复制必要的头文件以及共享库至相应的CUDA目录下,并赋予适当的读取权限给这些资源文件:
```bash
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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
```
#### 验证安装成果
最后可以编写一段简单的测试程序来确认CUDA是否正常工作。创建一个新的C++源码文件test.cu如下所示[^1]:
```cpp
#include <stdio.h>
int main() {
printf("Hello, CUDA!\n");
int deviceCount;
cudaError_t error = cudaGetDeviceCount(&deviceCount);
if (error != cudaSuccess){
fprintf(stderr,"CUDA Error: %s\n",cudaGetErrorString(error));
return 1;
}
printf("Found %d CUDA devices.\n", deviceCount);
return 0;
}
```
编译并执行这段代码即可查看是否有可用的GPU设备被检测出来:
```bash
nvcc test.cu -o test
./test
```
如果一切顺利的话,应该能看到有关已连接显卡数量的信息输出。
阅读全文
相关推荐


















