centos7安装cuda10.2和cudnn8
时间: 2023-10-03 19:03:36 浏览: 136
您好!要在CentOS 7上安装CUDA 10.2和cuDNN 8,您可以按照以下步骤进行操作:
1. 检查您的显卡是否支持CUDA 10.2,并确保您已经安装了适当的驱动程序。
2. 前往NVIDIA官网下载CUDA Toolkit 10.2的安装包(https://developer.nvidia.com/cuda-10.2-download-archive)。选择合适的版本(rpm或runfile)并下载到您的系统上。
3. 安装CUDA Toolkit 10.2。如果您下载的是rpm文件,请使用以下命令安装:
```
$ sudo rpm -i cuda-repo-<version>.rpm
$ sudo yum clean all
$ sudo yum install cuda
```
如果您下载的是runfile文件,请使用以下命令安装:
```
$ sudo sh cuda_<version>_linux.run
```
根据提示进行安装并接受许可协议。
4. 等待CUDA Toolkit的安装完成后,添加CUDA的环境变量。打开终端并编辑`~/.bashrc`文件,添加以下行:
```
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
```
保存文件后,运行`source ~/.bashrc`使其生效。
5. 下载cuDNN 8,您需要先注册NVIDIA开发者计划(如果尚未注册),然后前往https://developer.nvidia.com/rdp/cudnn-archive下载cuDNN 8对应的版本。
6. 将下载的cuDNN 8压缩文件解压缩,并将文件复制到CUDA的安装目录中。打开终端,运行以下命令:
```
$ tar -xzvf <path_to_cudnn_tar>
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda-10.2/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
$ sudo chmod a+r /usr/local/cuda-10.2/include/cudnn*.h /usr/local/cuda-10.2/lib64/libcudnn*
阅读全文