序文:
CUDAおよび cuDNN のインストール (Linux) Linuxにグラフィックス ドライバーと CUDA プログラムをインストールします。
システムパラメータ:
システム: ubuntu20.04.2
グラフィックカード: GeForce GTX 1050
インストールするバージョン:CUDA11.4
1.グラフィックカードドライバーをインストールします
(1) NVIDIA グラフィックス カードのモデルと推奨ドライバーを検出します。
ubuntu-drivers devices
(2) すべての推奨ドライバーのインストールを選択します
sudo ubuntu-drivers autoinstall
(3) nvidia-smi はドライバーのインストールが成功したかどうかを確認します
結果は次のことを示しています。
NVIDIA-SMI 470.74 ドライバー バージョン: 470.74 CUDA バージョン: 11.4
次に、CUDA プログラムをインストールします。
(1) cuda をダウンロードします。
NVIDIA と入力してインストール パッケージをダウンロードします。CUDA Toolkit 11.7 ダウンロード | NVIDIA 開発者
対応するバージョンを選択し、ターミナルに関連するコマンドを入力します (注: 独自のインストール バージョンを入力してください)。
wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
(2) cuda のインストール:
sudo sh cuda_11.0.2_450.51.05_linux.run
(3) cuda 環境変数の設定:
.1. vim パッケージをインストールします。
sudo apt install vim # version 2:8.1.2269-1ubuntu5.4, or
sudo apt install vim-tiny # version 2:8.1.2269-1ubuntu5.4
sudo apt install neovim # version 0.4.3-3
sudo apt install vim-athena # version 2:8.1.2269-1ubuntu5.4
sudo apt install vim-gtk3 # version 2:8.1.2269-1ubuntu5.4
sudo apt install vim-nox # version 2:8.1.2269-1ubuntu5.4
.2. 環境変数の設定 gedit ~/.bashrc
推荐用这个,打开是文件好操作
vim ~/.bashrc
ファイルを開いて次の内容を入力します: (注: i
插入 :
wq
退出
)
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
最後に、次のように入力して環境変数を更新します。
source ~/.bashrc
(4) インストールが成功したか確認する
nvcc -V
3. cudnn プログラムをインストールします
(1) cudnn ダウンロード
cuDNN は、ディープ ニューラル ネットワークの GPU アクセラレーション コンピューティング用のライブラリです。まずは公式サイトへ cuDNN Archive | NVIDIA Developer (2)インストール
解压缩->在压缩后的
cuda文件夹,打开终端
#注意: 解凍されたフォルダーは、対応するファイルを/usr/local の cuda にコピーします。
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*