Linux系統(tǒng)下CUDA和cuDNN環(huán)境配置方式
Linux下CUDA和cuDNN環(huán)境配置
本人配置環(huán)境:
- linux服務器
- ubantu18
- 顯卡驅(qū)動11.0
- 安裝CUDA11.0和cuDNN8.2.1
一、安裝CUDA11.0
1、先多找?guī)灼┛停私獯蟾诺牧鞒蹋苊獠瓤印?/p>
2、官網(wǎng)下載,復制官網(wǎng)命令到服務器下載。
CUDA Toolkit Archive | NVIDIA Developer
3、輸入命令,再按i鍵,進入vim的編輯模式。
sudo vim ~/.bashrc
4、在末尾添加如下兩行,注意里面的路徑要換成自己的版本,再按esc鍵退出vim編輯模式,并輸入 :wq 保存并退出。
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
5、以下命令激活配置,并使用nvcc -V查看配置效果,如圖表示成功。
source ~/.bashrc
二、安裝cuDNN8.2.1
1、官網(wǎng)下載和CUDA對應的版本
cuDNN Archive | NVIDIA Developer
2、解壓文件
tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
3、執(zhí)行如下四條命令,注意換成自己的版本。
sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.0/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64 sudo chmod a+r /usr/local/cuda-11.0/include/cudnn*.h sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*
4、使用如下命令查看cuDNN的版本號
cat /usr/local/cuda-11.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
總結(jié)
至此安裝完成。
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
CentOS7 + node.js + nginx + MySQL搭建服務器全過程
這篇文章主要介紹了關于CentOS7 + node.js + nginx + MySQL搭建服務器的全過程,文章通過一步步的步驟進行介紹的很詳細,相信對大家具有一定的參考價值,需要的朋友們下面來一起看看吧。2017-03-03sersync2 完全安裝配置說明(三) 插件基本配置和使用
sersync2 完全安裝配置說明(三) 插件基本配置和使用,需要的朋友可以參考下。2011-11-11CentOS7安裝調(diào)試Mysql數(shù)據(jù)庫的步驟詳解【實例】
這篇文章主要介紹了CentOS7安裝調(diào)試Mysql數(shù)據(jù)庫,本文分步驟通過實例代碼給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2019-10-10