Ubuntu 20.04 に upgrade 後 GPU まわりを修正2
Aim
PyTorch まわりでえらー?でるので 11.2→11.3に変更
tensorflow の対応が 11.2 までだが,cellpose で pytorch を使うにの cuda 11.3 と書いてあるので...
1. NVIDIA-driver, CUDA, CUDNN の uninstall
code: uninstall
sudo apt remove --purge nvidia*
sudo apt remove --purge cuda*
sudo apt remove --purge cudnn*
sudo apt remove --purge libnvidia
2. CUDA の install v11.3
code: install_cuda.sh
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda-11-3
#sudo apt-get -y install cuda-11-3 のほうが確実? bashrc に追加
code:bashrc
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
3. cuDNNのインストール
Download cuDNN v8.2.1 (June 7th, 2021), for CUDA 11.x を選択.
cuDNN Library for Linux (x86_64) をダウンロード
code:install-cudnn.sh
cd ~/Download
tar -xvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
sudo cp ./cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp -P ./cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
一応 test もしてみる.
code:test_cudnn.sh
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd ~/cudnn_samples_v8/mnistCUDNN/
make clean && make
#sudo apt install libfreeimage3 libfreeimage-dev ./mnistCUDNN
3-2 .deb をダウンロードして apt でいれてもおなじかな?
cuDNN Runtime Library for Ubuntu20.04 x86_64 (Deb)
cuDNN Developer Library for Ubuntu20.04 x86_64 (Deb)
cuDNN Code Samples and User Guide for Ubuntu20.04 x86_64 (Deb)
Remarks
Date :2023/01/13
Modified Date :20230418
Author :lattice.icon