ubuntu系统安装cuda(ubuntu18.04安装cuda和cudnn)
导读:一、安装 cuda...
一 、安装 cuda
1. 首先查看自身电脑最高支持的cuda版本为多少 ,在终端输入以下指令
nvidia-smi可以看到我的最高支持 cuda 11.4 。
2. 打开英伟达官网 下载官方cuda
此处我下载的是cuda 11.0.3 版本 ,并选择相应配置,复制指令至终端下载
3. 运行官网安装指令
4. 依次选择continue和accept 点击确认
5. 按回车取消Driver项的安装 ,然后回车选择Install
6. 添加环境变量 gedit ~/.bashrc打开文件 ,在文件结尾添加如下语句:
export PATH=$PATH:/usr/local/cuda-11.0/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64 export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib647. 保存 、更新环境变量
source ~/.bashrc8. 输入nvcc -V查看版本号
二、 安装 cudnn
1. 进入官网 ,寻找合适的 cudnn 版本 ,下载对应版本的cudnn 的 runtime developer 和samples
2. 下载得到四个文件
3. 安装上述软件包
tar -zxvf cudnn-11.0-linux-x64-v8.0.4.30.tgz sudo cp -rf cuda/include/cudnn* /usr/local/cuda/include/ sudo cp -rf cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn* sudo chmod a+r /usr/local/cuda/lib64/libcudnn* sudo dpkg -i libcudnn8_8.0.4.30-1+cuda11.0_amd64.deb sudo dpkg -i libcudnn8-dev_8.0.4.30-1+cuda11.0_amd64.deb sudo dpkg -i libcudnn8-samples_8.0.4.30-1+cuda11.0_amd64.deb4. 测试
拷贝例程到用户目录
cp -r /usr/src/cudnn_samples_v8/ $HOME5. 转到minist例程
cd $HOME/cudnn_samples_v8/mnistCUDNN6. 编译例程
sudo make clean && make7. 运行例程
./mnistCUDNN8. 如果cudnn被恰当的安装 ,将输出打印消息:Test passed!
9. cudnn 检测版本号
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!