1.CUDAのインストール
- CUDA Toolkit 11.6 Downloadsより、Select Target Platformで自分に合った情報を選択する
- 筆者の場合はUbuntu20.04などを使っているため、以下の情報を選択
- インストール手順のコマンドの一覧
- 手順通りにコマンドを実行
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
sudo apt-get -y install cudaをsudo apt-get -y install cuda-driversとすると、最新のドライバだけ綺麗にインストールができます。
2.NVIDIAドライバのインストール
-
Installation GuideのSetting up NVIDIA Container Toolkit以下のコマンドを実行
-
インストール手順のコマンド
- 手順通りにコマンドを実行
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker
3.再起動し、インストールされているか確認
- 以下のコマンドを実行し、画像のような出力ができるか確認
nvidia-smi
4.もしも再起動後に画面が真っ暗になった時の対処法
- 以下のサイトの手順通りにやれば解決する
- [Ubuntu]CUDAを入れたら画面が真っ黒になった話