centos7.8安装cuda10.1
时间: 2023-08-09 20:07:20 浏览: 94
安装CUDA10.1和Cudnn所需的步骤如下:
1. 安装CUDA10.1:
a) 下载安装包:从NVIDIA官网下载适合自己系统的CUDA10.1安装包。
b) 安装CUDA:使用命令行或图形界面按照CUDA安装向导完成CUDA10.1的安装。
c) 设置环境变量:在.bashrc文件中添加以下环境变量:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
2. 安装Cudnn:
a) 下载Cudnn:从NVIDIA官网下载适合自己系统的Cudnn安装包。
b) 安装Cudnn:使用命令行或图形界面按照Cudnn安装向导完成Cudnn的安装。
c) 设置环境变量:在.bashrc文件中添加以下环境变量:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
3. 检查安装:
a) 使用命令行检查CUDA版本:输入“nvcc -V”,如果输出正常则表示CUDA安装成功。
b) 使用命令行检查Cudnn版本:输入“cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR”,如果输出正常则表示Cudnn安装成功。
安装完毕后,即可开始进行深度学习等相关任务的工作。
阅读全文