すっぴんのUbuntu 20.04 LTSに最短でCUDA入れるメモ(CUDA11.6編)

前回の一部改変

bleu48.hatenablog.com

 

すっぴんのUbuntu 20.04 LTSにCUDA11.6入れるメモ

(所謂自分用コピペ元)

 

まず以下を確認

https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/

 

sudo wget -O /etc/apt/preferences.d/cuda-repository-pin-600 https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"

 

sudo apt -y install cuda-11-6

 

export CUDA_PATH=/usr/local/cuda-11.6
echo 'export CUDA_PATH=/usr/local/cuda-11.6' >> ${HOME}/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:${LD_LIBRARY_PATH}
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:${LD_LIBRARY_PATH}' >> ${HOME}/.bashrc
export PATH=/usr/local/cuda-11.6/bin:${PATH}
echo 'export PATH=/usr/local/cuda-11.6/bin:${PATH}' >> ${HOME}/.bashrc

 

以下は機械学習系(TensorRTなど)

 

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/ /"

 

sudo apt-get install libnvinfer-dev libnvonnxparsers-dev libnvparsers-dev libnvinfer-plugin-dev

sudo apt-get install zip unzip

 

---

もちろん,5月の選手権で使うCUDAのバージョンが11.6ということです。

昨年一次予選1位,二次予選11位で第2回電竜戦予選1位,決勝4位の二番絞りの進化を御期待下さい。