linux安装cuda11.8,cudnn
时间: 2024-11-18 15:04:55 浏览: 83
在Linux系统上安装CUDA 11.8和cuDNN通常需要几个步骤:
1. **下载**:
- 首先访问NVIDIA官网 (https://developer.nvidia.com/cuda-downloads) 下载适用于你的GPU型号和Linux发行版的CUDA 11.8 SDK。
- 同时,你也需要cuDNN库,从NVIDIA Developer网站上下载cuDNN 8.0版本,因为CUDA 11.8支持cuDNN 8.x。
2. **添加存储库**:
- 对于Ubuntu或Debian系统,你可以通过添加NVIDIA官方存储库来简化安装过程,命令类似:
```
sudo add-apt-repository ppa:graphics-drivers/ppa
```
3. **更新软件包列表**:
```
sudo apt-get update
```
4. **安装CUDA**:
```
sudo apt-get install cuda=11.8.*
```
这会安装CUDA工具链、驱动和其他必要的组件。
5. **安装cuDNN**:
- 解压下载的cuDNN文件,通常会有`cudnn*`文件夹,然后将`lib64`目录下所有内容复制到`/usr/local/cuda/lib64/`。
- 创建软链接指向cuDNN的头文件和库:
```
sudo cp cudnn/include/cudnn.h /usr/local/cuda/include/
sudo ln -s /usr/local/cuda/include/cudnn.h /usr/include/cudnn.h
sudo cp -r cudnn/lib64/* /usr/local/cuda/lib64/
sudo ln -s /usr/local/cuda/lib64/libcudnn* /usr/local/cuda/lib64/
```
6. **配置环境变量**:
添加CUDA路径到`LD_LIBRARY_PATH`和`PATH`:
```
echo "export CUDA_HOME=/usr/local/cuda" >> ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
source ~/.bashrc
```
7. **验证安装**:
使用`nvcc --version`检查CUDA编译器,运行`nvidia-smi`查看GPU状态,以及`cudnnTest`或`cupti_test`来验证cuDNN是否成功安装。
阅读全文