CUD11.1 下載地址:CUDA Toolkit Archive | NVIDIA Developer
?安裝:
wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run
對于不是sudo用戶,可以不執(zhí)行sudo,不過沒辦法裝到/usr/local/,可以裝到你有權(quán)限的文件夾目錄。
裝完后,需要增加環(huán)境路徑到~/.bashrc并source
# cuda enviroment
export PATH="/home/xgg/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/home/xgg/cuda-11.1/lib64:$LD_LIBRARY_PATH"
export CPATH=/home/xgg/cuda-11.1/include:$CPATH
?記得source ~/.bashrc使得環(huán)境生效
cuDNN8.6.0,這個版本我是根據(jù)我需要安裝的TensorRT的版本要求下的,比如你要下載8.5.3的TensorRT,下載地址:https://developer.nvidia.com/nvidia-tensorrt-8x-download
可以看見文件末尾會有個cuDNN版本,這個是推薦版本
所以我下載8.6.0的cuDNN,下載地址:cuDNN Archive | NVIDIA Developer
下載后我的是tar.xz格式,需要兩步驟解壓
①xz -d cudnn*.tar.xz解壓成.tar
②tar xvf cudnn*.tar解壓成文件夾
將解壓好后的文件夾拷貝到CUDA文件夾中
# cd到你解壓的cudnn文件夾
cd cudnn*
# 拷貝lib到cuda的lib中
cp lib/* /你的cuda路徑/cuda-11.1/lib64/
# 拷貝include到cuda的include中
cp include/* /你的cuda路徑/cuda-11.1/include/
最后記得將拷貝過去的相關(guān)文件夾賦予權(quán)限
chmod a+r /你的cuda路徑/cuda-11.1/include/cudnn*
chmod a+r /你的cuda路徑/cuda-11.1/lib64/libcudnn*
驗證cuDNN版本與安裝信息
cat /你的cuda路徑/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
# 或者如下,一般是上面的命令
cat /你的cuda路徑/cuda-11.1/include/cudnn.h | grep CUDNN_MAJOR -A 2
TensorRT8.5.3,下載地址:https://developer.nvidia.com/nvidia-tensorrt-8x-download
下載后解壓,解壓命令:tar -zxvf Tensor*.tar.gz
解壓后需要在~/.bashrc中添加環(huán)境路徑
# tensorrt environment
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/你的TensorRT路徑/TensorRT-8.5.3.1/lib
export CPATH=/你的TensorRT路徑/TensorRT-8.5.3.1/include:$CPATH
?基本裝好了,然后可以把文件夾中附帶的軟件都裝了
?以Python舉例,cd到python中
選擇你得python版本對應的包,pip install安裝就行。文章來源:http://www.zghlxwxcb.cn/news/detail-551132.html
以上就是這些,之后有問題我再來補充文章來源地址http://www.zghlxwxcb.cn/news/detail-551132.html
到了這里,關(guān)于CUDA11.1、cuDNN8.6.0、Tensorrt8.5.3,ubuntu20.04安裝過程記錄的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!