ubuntu20.04+cuda+cudnnのインストール

序文:

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*

おすすめ

転載: blog.csdn.net/m0_48919875/article/details/125161788